flax | Python Package Wiki

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

gin-config	Gin-Config: A lightweight configuration library for Python
dm-acme	A Python library for Reinforcement Learning.
tensorflow-ranking	Pip package setup file for TensorFlow Ranking.
tf-slim	TensorFlow-Slim: A lightweight library for defining, training and evaluating complex models in TensorFlow
tf-agents-nightly	TF-Agents: A Reinforcement Learning Library for TensorFlow
dm-reverb-nightly	Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research.
trfl	trfl is a library of building blocks for reinforcement learning algorithms.
glfw	A ctypes-based wrapper for GLFW3.
tensorflow-datasets	tensorflow/datasets is a library of datasets ready to use with TensorFlow.
tensorflow-estimator-2.0-preview	TensorFlow Estimator.
tf-estimator-nightly	TensorFlow Estimator.
sacrebleu	Hassle-free computation of shareable, comparable, and reproducible BLEU, chrF, and TER scores
absl-py	Abseil Python Common Libraries, see https://github.com/abseil/abseil-py.
tfp-nightly	Probabilistic modeling and statistical inference in TensorFlow
tb-nightly	TensorBoard lets you watch Tensors Flow
tensor2tensor	Tensor2Tensor
tensorflow-probability	Probabilistic modeling and statistical inference in TensorFlow
dm-env	A Python interface for Reinforcement Learning environments.
gym	The OpenAI Gym: A toolkit for developing and comparing your reinforcement learning agents.

`pip install flax==0.3.6`