luchris429

purejaxrl

Really Fast End-to-End Jax RL Implementations

popjaxrl

Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]

model-free-opponent-shaping

Code for Model-Free Opponent Shaping (ICML 2022)

discovered-policy-optimisation

Code for Discovered Policy Optimisation (NeurIPS 2022)

adversarial-cheap-talk

Code for Adversarial Cheap Talk (ICML 2023)

Awesome-Evolutionary-Reinforcement-Learning

Research Papers and Code Repository on the Integration of Evolutionary Algorithms and Reinforcement Learning

hackjam_hackathon

noisy-tv-env

The Noisy TV Environment from "Large-Scale Study of Curiosity-Driven Learning" (ICLR 2019)

awesome-jax

JAX - A curated list of resources https://github.com/google/jax

dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

gymnax

RL Environments in JAX 🌍

ml-agents

Unity Machine Learning Agents

nanoGPT

The simplest, fastest repository for training/finetuning medium-sized GPTs.

pagesTest

swe_behavioral_cloning

trimesh

Python library for loading and using triangular meshes.