Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Here you can share your experience with the project you are suggesting or its comparison with jax. Optional.
A valid email to send you a verification link when necessary or log in.
jax