absl-py
clu
dataclasses
flax
gym
grpcio
jax
jaxlib
numpy
optax
tensorflow_probability[jax]

[develop]
pytest
transforms3d
