carlosgmartin

nash-finder

Find Nash equilibrium for all games

nushell

A new type of shell

fax

typed-argument-parser

Typed argument parser for Python

equivariant-MLP

A library for programmatically generating equivariant layers through constraint solving

stockfish

Integrates the Stockfish chess engine with Python (Official fork)

Teg

A differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and discontinuities.

multiplayer-alphazero

PyTorch AlphaZero implementation with multiplayer support [NeurIPS 2019 Deep Reinforcement Learning Workshop]

h-levin

Levin tree search guided by both a policy and a heuristic function

prodigy

The Prodigy optimizer and its variants for training neural networks.

muax

A project that provides help for using DeepMind's mctx on gym-style environments.

mctx-az

Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree

aleph_star

Reinforcement learning with A* and a deep heuristic

JaxMARL

Multi-Agent Reinforcement Learning with JAX

stable-opponent-shaping

Pytorch implementation of Stable Opponent Shaping (https://openreview.net/pdf?id=SyGjjsC5tQ).

tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

cursor

The AI Code Editor

cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

purejaxrl

Really Fast End-to-End Jax RL Implementations

awesome-ml4co

Awesome machine learning for combinatorial optimization papers.

google-research

Google Research

uncertainty-baselines

High-quality implementations of standard and SOTA methods on a variety of tasks.

pipes

Pipelines for expressive code on collections in C++

alpa

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

gym-sokoban

Sokoban environment for OpenAI Gym

mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:

nengolib

Nengo library of additional extensions

awesome-jax

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

gymnax

RL Environments in JAX 🌍

jaxnet

Concise deep learning for JAX