Comment Comment by alexbw Check out github.com/google/jax, it’s NumPy on the GPU with automatic differentiation, JIT and autobatching. parent alexbw 8 years ago
