JAX is a research project that grew out of Autograd. Here's a two-page abstract about an early version. Watch this space for updates!
This is not an official Google product.
forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 0
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
License
jjwilke/jax
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 92.5%
- C++ 4.8%
- Jupyter Notebook 1.3%
- Starlark 1.1%
- Shell 0.2%
- C 0.1%