Top jax issues

Add support for Jax on Windows

build contributions welcome good first issue
dwang55 picture dwang55·23 Feb 2019·22 Comments
👍10 ❤1 😕1

BFGS/Quasi-Newton optimizers?

proteneer picture proteneer·26 Sep 2019·53 Comments
👍21

How to print values inside a @jit-compiled function?

question
ndronen picture ndronen·5 Jan 2019·4 Comments
👍2

Error "UnboundLocalError: local variable 'name' referenced before assignment" import jax.numpy as np

build
tranhoangkhuongvn picture tranhoangkhuongvn·17 Jan 2019·5 Comments

cannot find libdevice

murphyk picture murphyk·8 Jul 2019·31 Comments

conda-based installation

build contributions welcome enhancement
ericmjl picture ericmjl·3 Jan 2019·53 Comments
❤4

Add support for AMD GPUs

contributions welcome
ricardobarroslourenco picture ricardobarroslourenco·16 Jan 2020·35 Comments

Add support for buffer donation (input/output aliasing)

enhancement
hawkinsp picture hawkinsp·21 Nov 2019·19 Comments
🚀10 👍6

Installing from source using Conda and CUDA could be improved

build
kirk86 picture kirk86·1 Feb 2019·22 Comments

`jaxlib==0.1.44` segfaults when trying to run XLA on GPU

bug
ziyadedher picture ziyadedher·17 Apr 2020·15 Comments
Next issues