C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.

⚡ Flashbax: Accelerated Replay Buffers in JAX

Massively parallel rigidbody physics simulation on accelerator hardware.

MLX: An array framework for Apple silicon

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

♟️ Vectorized RL game environments in JAX

A Library for Uncertainty Quantification.

A playbook for systematically maximizing the performance of deep learning models.

A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

RL Environments in JAX 🌍

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Monte Carlo tree search in JAX

Scenic: A Jax Library for Computer Vision Research and Beyond

Rich is a Python library for rich text and beautiful formatting in the terminal.

Modular framework for Reinforcement Learning in python

Simple, powerful, and fast logging for Python.

Python Classes Without Boilerplate

ML Collections is a library of Python Collections designed for ML use cases.

Matplotlib styles for scientific plotting

Containers for machine learning

📊 Save matplotlib figures as TikZ/PGFplots for smooth integration into LaTeX.

Python logging made (stupidly) simple

Convert LaTeX documents into beautiful responsive web pages using LaTeXML.

Version control for machine learning

arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv

Gin provides a lightweight configuration framework for Python

Modin: Scale your Pandas workflows by changing a single line of code

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

A Hyperparameter Tuning Library for Keras

A library of reinforcement learning components and agents