Top jax issues

JAX_ENABLE_X64 norm.cdf inconsistent with scipy

bug
JuliusKunze picture JuliusKunze·8 Jul 2019·8 Comments

inf breaks scipy.stats.norm.cdf for array size >= 4

bug
JuliusKunze picture JuliusKunze·3 Jul 2019·2 Comments

pmap is pretty slow for functions involving jax.random

fehiepsi picture fehiepsi·7 Jun 2019·8 Comments

scan abstract values and tree prefix convention bug

bug
mattjj picture mattjj·30 May 2019·5 Comments
👍1

Please enable vmap(scan())

enhancement
sussillo picture sussillo·15 May 2019·3 Comments
👍5

segment_sum primitives / advanced indexing in jax.ops.index_add

enhancement
proteneer picture proteneer·30 Apr 2019·5 Comments
👍1

controlling float precision?

documentation
jkitchin picture jkitchin·29 Apr 2019·2 Comments

Discrepancy with autograd results

madvn picture madvn·10 Apr 2019·3 Comments

`np.sum` lost precision for large array

fehiepsi picture fehiepsi·8 Apr 2019·17 Comments

Performance in small functions

ColCarroll picture ColCarroll·7 Apr 2019·8 Comments
👍1
 Previous issues
Next issues