Top jax issues

Improve the speed of the first step of while_loop/fori_loop

fehiepsi picture fehiepsi·8 Apr 2019·4 Comments

Implementing Linear Regression in jax

question
ghost picture ghost·6 Apr 2019·8 Comments

Calling a jit'd function with or without kwarg name has different behavior

enhancement
skye picture skye·22 Mar 2019·10 Comments

Time-efficient higher-order forward mode?

enhancement question
duvenaud picture duvenaud·20 Mar 2019·18 Comments
👍2

Document `lax.stop_gradient`

documentation
act65 picture act65·10 Mar 2019·1 Comments

[Question] Efficiently sampling from the binomial distribution

neerajprad picture neerajprad·6 Mar 2019·4 Comments
👍1

np.exp does not return inf on overflow in fast math mode

bug
fehiepsi picture fehiepsi·2 Mar 2019·8 Comments

Optimizing multiple parameters via jax optimizer

neerajprad picture neerajprad·20 Feb 2019·2 Comments

jax missing scipy.special.expit

asross picture asross·5 Feb 2019·3 Comments
👍1

add complex128 support (and match NumPy behavior)

enhancement
sjperkins picture sjperkins·10 Jan 2019·9 Comments
 Previous issues
Next issues