8bitmp3

AutoROM

A tool to automate installing Atari ROMs for the Arcade Learning Environment

Gymnasium-Robotics

A collection of robotics simulation environments for reinforcement learning

PettingZoo

An API standard for multi-agent reinforcement learning environments, with popular reference environments and related utilities

Doom-Reinforcement-Learning-Models

Reinforcement Learning / CNN model trained to play Doom video game within the VizDoom gym environment, for a variety of scenarios and difficulties.

VizDoomGym

Highly Customizable Gym interface for ViZDoom.

ViZDoom

Reinforcement Learning environments based on the 1993 game Doom :godmode:

flax

Flax is a neural network library for JAX that is designed for flexibility.

pyink

Pyink, pronounced pī-ˈiŋk, is a Python formatter, forked from Black with a few different formatting behaviors.

realbook

Easier audio-based machine learning with TensorFlow.

mangle

jax-streams

ml_dtypes

A stand-alone implementation of several NumPy dtype extensions used in machine learning.

jax-tpu-embedding

tf-aoc

Advent of Code in pure TensorFlow

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

australis

jax-cosmo-paper

Paper introducing jax-cosmo

jaxDecomp

JAX bindings for the NVIDIA cuDecomp library

minGPT

minGPT in JAX

clip-jax

Train vision models using JAX and 🤗 transformers

jax-meta-learning

A collection of meta-learning algorithms in Jax

jbdl

A JAX-Based Body Dynamics Algorithm Library for Robotics.

jaxga

Geometric Algebra package for JAX

jax_muzero

An implementation of MuZero in JAX.

a0-jax

AlphaZero in JAX

jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

imax

Image augmentation library for Jax

ShinRL

ShinRL: A Library for Evaluating RL Algorithms from Theoretical and Practical Perspectives (Deep RL Workshop 2021)

efax

Exponential families for JAX

alpa

Training and serving large-scale neural networks with auto parallelization.