ScottAlexanderCameron / Jynx
A neural network library written in jax
☆11Updated 3 weeks ago
Alternatives and similar repositories for Jynx:
Users that are interested in Jynx are comparing it to the libraries listed below
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆216Updated last week
- A tool for aggregating and plotting MARL experiment data.☆64Updated last month
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆251Updated last week
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆50Updated last year
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆216Updated 2 months ago
- Benchmarking RL generalization in an interpretable way.☆134Updated 10 months ago
- Accelerated Quality-Diversity☆270Updated 2 months ago
- A collection of matrix games in JAX☆10Updated 2 weeks ago
- ☆158Updated this week
- 🐈⬛ Contextual bandits library for continuous action trees with smoothing in JAX☆65Updated 2 years ago
- Partially Observable Process Gym☆168Updated 5 months ago
- ☆68Updated last month
- Datasets with baselines for offline multi-agent reinforcement learning.☆152Updated last month
- JAX implementation of RL algorithms and vectorized environments☆37Updated 11 months ago
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated 11 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆109Updated 2 weeks ago
- A framework for Reinforcement Learning research.☆128Updated 2 weeks ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆127Updated 2 weeks ago
- ☆65Updated 3 months ago
- ☆18Updated 4 months ago
- ☆288Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆100Updated 2 years ago
- ☆206Updated 3 weeks ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆118Updated last year
- JAX implementations of various deep reinforcement learning algorithms.☆21Updated 3 years ago
- ☆39Updated last year
- ReDo: The Dormant Neuron Phenomenon in Deep Reinforcement Learning (pytorch)☆25Updated last month
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆47Updated last year