Skip to content
/ jax Public
forked from jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

License

Notifications You must be signed in to change notification settings

ROCm/jax

Error
Looks like something went wrong!

About

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 85.6%
  • C++ 11.9%
  • Starlark 1.7%
  • Jupyter Notebook 0.4%
  • Shell 0.3%
  • C 0.1%