Alternatives and similar repositories for granular

Users that are interested in granular are comparing it to the libraries listed below

• danijar / teleport
  Efficiently send large arrays across machines
  ☆16Updated 10 months ago
• danijar / elements
  Building blocks for productive research
  ☆55Updated 4 months ago
• danijar / ninjax
  General Modules for JAX
  ☆66Updated 2 months ago
• arnavkj1995 / SFM
  Official Implementation of SFM and the baselines in Jax.
  ☆18Updated last week
• danijar / portal
  Fast and reliable distributed systems in Python
  ☆28Updated last month
• ethanluoyc / corax
  Corax: Core RL in JAX
  ☆38Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆57Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 7 months ago
• microsoft / segar
  Sandbox environment for generalizable agent research
  ☆25Updated 2 years ago
• gauthamvasan / avg
  Action Value Gradient Algorithm
  ☆20Updated 2 weeks ago
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆28Updated this week
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 10 months ago
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆20Updated 11 months ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆10Updated last year
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆21Updated 2 years ago
• AutonomousAgentsLab / cr-dv3
  DreamerV3 implementation of Curious Replay, a method for prioritizing experience replay that is tailored to model-based reinforcement lea…
  ☆36Updated last year
• facebookresearch / controllable_agent
  The Controllable Agent project trains RL Agents able to optimize any reward function specified in real time, without any further learning…
  ☆62Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 4 months ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆18Updated 2 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆29Updated last year
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated last year
• radarFudan / mamba-minimal-jax
  ☆29Updated 6 months ago
• google-deepmind / csuite
  ☆44Updated 8 months ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated 9 months ago
• nmonette / NCC-UED
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆11Updated last week
• yuqingd / cusp
  ☆15Updated 2 years ago
• notmahi / disk
  PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…
  ☆19Updated 3 years ago
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆26Updated 8 months ago