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â228Updated last week
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ304Updated this week
- A tool for aggregating and plotting MARL experiment data.â76Updated 2 months ago
- â191Updated 3 months ago
- Clean single-file implementation of offline RL algorithms in JAXâ138Updated 3 months ago
- Accelerated Quality-Diversityâ282Updated last week
- Benchmarking RL generalization in an interpretable way.â151Updated 2 weeks ago
- Collection of resources on plasticity loss in deep reinforcement learningâ15Updated 4 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ150Updated last week
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ57Updated last year
- â73Updated 4 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"â100Updated 2 years ago
- Partially Observable Process Gymâ183Updated 8 months ago
- đââŦ Contextual bandits library for continuous action trees with smoothing in JAXâ66Updated 2 years ago
- â218Updated 4 months ago
- Accelerated minigrid environments with JAXâ132Updated 7 months ago
- Goal-Conditioned Reinforcement Learning with JAXâ134Updated this week
- Gridworld domains in the gym interfaceâ27Updated 5 months ago
- Author's PyTorch implementation of TD7 for online and offline RLâ139Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023â19Updated last year
- A Simplified Pytorch Version of the Dreamer Algorithmâ123Updated last year
- Datasets with baselines for offline multi-agent reinforcement learning.â162Updated this week
- â298Updated 3 months ago
- â18Updated 2 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ424Updated 2 weeks ago
- JAX implementation of RL algorithms and vectorized environmentsâ42Updated last year
- The Starcraft Multi-Agent challenge liteâ42Updated 6 months ago
- A collection of RL algorithms written in JAX.â96Updated 2 years ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ709Updated this week
- Conservative Q Learning on top of SACâ129Updated 2 years ago