GitHub
jax (Frameworks)
JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. | Python | ⭐ 35,000 | Frameworks
1 articles in archive
[Frameworks] jax-ml/jax — 35.0k⭐
JAX is a Python library for accelerator-oriented array computation and program transformation, designed for high-performance numerical computing and large-scale machine learning. Language: Python | Stars: 35.0k | Category: Frameworks Repository: https://github.com/jax-ml/jax
GitHub1d ago