optax | Python Package Wiki

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

cloud-volume	A serverless client for reading and writing Neuroglancer Precomputed volumes both locally and on cloud services.
tf-agents	TF-Agents: A Reinforcement Learning Library for TensorFlow
baselines	OpenAI baselines: high quality implementations of reinforcement learning algorithms
pycolab	An engine for small games for reinforcement learning agents.
tensorboard_logger	Log TensorBoard events without Tensorflow
arf	Advanced Recording Format for acoustic, behavioral, and physiological data
face_recognition_models	Models used by the face_recognition package.
pyglib	Opinionated but handy app startup wrapper.
tflite-support	TFLite Support is a toolkit that helps users to develop ML and deploy TFLite models onto mobile devices.
dm-env	A Python interface for Reinforcement Learning environments.
hachoir	Package of Hachoir parsers used to open binary files
cleverhans	None
chainer-libdnn	General utilities for neural networs with chainer
valerius	Biological sequence analysis tools.
rmsd	Calculate root-mean-square deviation (RMSD) between two sets of cartesian coordinates (XYZ or PDB format), using rotation (fx. Kabsch algorithm), atom reordering (fx. Hungarian algorithm), and axis reflections, resulting in the minimal RMSD.
dyNET	The Dynamic Neural Network Toolkit
pyrouge	A Python wrapper for the ROUGE summarization evaluation package.
alibi	Algorithms for monitoring and explaining machine learning models
diversipy	Sample in hypercubes, select diverse subsets, and measure diversity

`pip install optax==0.0.9`