jaxlib | Python Package Wiki

Based on how often these packages appear together in public requirements.txt files on GitHub.

jax	Differentiate, compile, and transform Numpy code.
tf-nightly	TensorFlow is an open source machine learning framework for everyone.
tensorflow-cpu	TensorFlow is an open source machine learning framework for everyone.
gin-config	Gin-Config: A lightweight configuration library for Python
onnx	Open Neural Network Exchange
cma	CMA-ES, Covariance Matrix Adaptation Evolution Strategy for non-linear numerical optimization in Python
gym	The OpenAI Gym: A toolkit for developing and comparing your reinforcement learning agents.
tfp-nightly	Probabilistic modeling and statistical inference in TensorFlow
tensorflow-addons	TensorFlow Addons.
opencv-python	Wrapper package for OpenCV python bindings.
imgaug	Image augmentation library for deep neural networks
thunder-registration	algorithms for image registration
deepdish	Deep Learning experiments from University of Chicago.
tensorflow-probability	Probabilistic modeling and statistical inference in TensorFlow
fuel	Data pipeline framework for machine learning
pymultinest	Access modules for the MultiNest and Cuba integration libraries
seqeval	Testing framework for sequence labeling
pybullet	Official Python Interface for the Bullet Physics SDK specialized for Robotics Simulation and Reinforcement Learning
lsanomaly	Least squares anomaly detection.

Proportion of downloaded versions in the last 3 months (only versions over 1%).

pip install jaxlib==0.1.73