Top jax issues

Very slow CPU + GPU compilation for large-ish model

j-towns picture j-towns·26 Mar 2019·6 Comments
👍2

[FR] support batching rule for `while`

enhancement
fehiepsi picture fehiepsi·25 Feb 2019·11 Comments

jax.scipy.linalg routines segfault on Mac OS X on scipy 1.2.1 or later but not scipy 1.1.0

bug
cdgreenidge picture cdgreenidge·22 Feb 2019·10 Comments
👍1

Default XLA/GPU memory allocator is synchronous/slow.

enhancement performance
hawkinsp picture hawkinsp·20 Feb 2019·4 Comments

Clarify: Larger-than-memory arrays

question
bionicles picture bionicles·16 Feb 2019·2 Comments

dev timeline for more batched ops

enhancement
marikgoldstein picture marikgoldstein·3 Feb 2019·5 Comments

Unimplemented NumPy core functions

enhancement good first issue
alexbw picture alexbw·11 Dec 2018·26 Comments

Keras NumPy backend demo

alexbw picture alexbw·11 Dec 2018·4 Comments
👍1

Frequenty asked questions doc

alexbw picture alexbw·11 Dec 2018·11 Comments
👍5

utility for closure-conversion in higher-order functions with custom derivatives

enhancement
froystig picture froystig·18 Dec 2020·7 Comments
 Previous issues
Next issues