pip install jax==0.2.24
Differentiate, compile, and transform Numpy code.
SourceAmong top 1000 packages on PyPI.
Over 5.2M downloads in the last 90 days.
jax
Based on how often these packages appear together in public
requirements.txt
files on GitHub.
XLA library for JAX |
|
TensorFlow is an open source machine learning framework for everyone. |
|
Gin-Config: A lightweight configuration library for Python |
|
Probabilistic modeling and statistical inference in TensorFlow |
|
TensorFlow is an open source machine learning framework for everyone. |
|
Probabilistic modeling and statistical inference in TensorFlow |
|
The OpenAI Gym: A toolkit for developing and comparing your reinforcement learning agents. |
|
Wrapper package for OpenCV python bindings. |
|
tensorflow/datasets is a library of datasets ready to use with TensorFlow. |
|
TensorFlow is an open source machine learning framework for everyone. |
|
Abseil Python Common Libraries, see https://github.com/abseil/abseil-py. |
|
Flax: A neural network library for JAX designed for flexibility |
|
TensorFlow Addons. |
|
Pure python package for DICOM medical file reading and writing |
|
Hassle-free computation of shareable, comparable, and reproducible BLEU, chrF, and TER scores |
|
TensorFlow is an open source machine learning framework for everyone. |
|
Access modules for the MultiNest and Cuba integration libraries |
|
Official Python Interface for the Bullet Physics SDK specialized for Robotics Simulation and Reinforcement Learning |
|
CMA-ES, Covariance Matrix Adaptation Evolution Strategy for non-linear numerical optimization in Python |
jax
Proportion of downloaded versions in the last 3 months (only versions over 1%).
0.2.24 |
54.39% |
0.2.19 |
15.42% |
0.2.20 |
6.84% |
0.2.21 |
6.55% |
0.2.18 |
4.83% |
0.2.17 |
3.29% |
0.2.22 |
1.88% |
0.2.16 |
1.81% |
0.2.10 |
1.72% |
0.2.12 |
1.38% |