ScottAlexanderCameron / Jynx
A neural network library written in jax
☆11Updated last month
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☆219Updated this week
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆263Updated 2 weeks ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆51Updated last year
- Benchmarking RL generalization in an interpretable way.☆138Updated 11 months ago
- A tool for aggregating and plotting MARL experiment data.☆68Updated this week
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆227Updated 3 months ago
- 🐈⬛ Contextual bandits library for continuous action trees with smoothing in JAX☆65Updated 2 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆121Updated 3 weeks ago
- A collection of matrix games in JAX☆10Updated last month
- ☆71Updated 2 months ago
- ☆175Updated last month
- Accelerated Quality-Diversity☆273Updated 3 months ago
- A Simplified Pytorch Version of the Dreamer Algorithm☆114Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆19Updated last year
- Gridworld domains in the gym interface☆27Updated 3 months ago
- Partially Observable Process Gym☆170Updated 6 months ago
- ReDo: The Dormant Neuron Phenomenon in Deep Reinforcement Learning (pytorch)☆25Updated 2 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆113Updated last week
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆121Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆102Updated 2 years ago
- Conservative Q Learning on top of SAC☆122Updated 2 years ago
- Representation Learning for RL☆121Updated last year
- ☆293Updated 3 weeks ago
- ☆18Updated last week
- JAX implementations of various deep reinforcement learning algorithms.☆21Updated 3 years ago
- Datasets with baselines for offline multi-agent reinforcement learning.☆157Updated 2 months ago
- Skeleton for scalable and flexible Jax RL implementations☆67Updated last year
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆98Updated 7 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆132Updated last month
- JAX implementation of RL algorithms and vectorized environments☆38Updated last year