pip install jaxlib==0.1.73

XLA library for JAX

Source
Among top 1000 packages on PyPI.
Over 2.9M downloads in the last 90 days.

Commonly used with jaxlib

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.

Version usage of jaxlib

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

0.1.70

32.84%

0.1.69

30.01%

0.1.71

25.30%

0.1.73

3.41%

0.1.72

3.10%

0.1.62

2.70%