Package profile
jax
- Summary: Differentiate, compile, and transform Numpy code.
- Author: JAX team
- License: Apache-2.0
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax
- Number of releases: 198
- First release: 0.0 on 2018-12-07
- Latest release: 0.12.0.1 on 2026-10-21
- Latest release size: 3.3 MB (pure Python wheel)
Appears in:The Python packages with the highest Wikipedia word countsThe Python packages with the most Wikipedia pages
Dependencies
Jax has 13 dependencies, 8 of which optional.Dependent packages
| Package | Optional | Group |
|---|---|---|
| brax | false | |
| chex | false | |
| clu | false | |
| distrax | false | |
| dopamine-rl | false |
Similar packages
- autogradEfficiently computes derivatives of NumPy code.
- nptypingType hints for NumPy.
- numexprFast numerical expression evaluator for NumPy
- array-api-compatA wrapper around NumPy and other array libraries to make them compatible with the Array API standard
- numpy-quaternionAdd a quaternion dtype to NumPy
- uprootROOT I/O in pure Python and NumPy.
- bottleneckFast NumPy array functions written in C