subho406 / agalite

Related projects ⓘ

Alternatives and complementary repositories for agalite

• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 2 years ago
• radarFudan / mamba-minimal-jax
  ☆29Updated 2 months ago
• Doraemonzzz / hgru-pytorch
  ☆25Updated 4 months ago
• keraJLi / synthetic-gymnax
  ☆17Updated 5 months ago
• subho406 / Recurrent-PPO-Jax
  Implementation of Proximal Policy Optimization in Jax+Flax
  ☆16Updated last year
• abhayraw1 / planet-torch
  A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning
  ☆11Updated 4 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆40Updated 2 years ago
• arnavkj1995 / VSG
  Learning Robust Dynamics Through Variational Sparse Gating
  ☆20Updated 2 years ago
• FLAIROx / cultural-accumulation
  ☆13Updated 4 months ago
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 2 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 2 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• chandar-lab / Recall2Imagine
  Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024
  ☆50Updated 5 months ago
• ahjwang / messenger-emma
  Implements the Messenger environment and EMMA model.
  ☆23Updated last year
• chongyi-zheng / td_infonce
  ☆25Updated last year
• danijar / ninjax
  General Modules for JAX
  ☆58Updated 3 months ago
• Cranial-XIX / longhorn
  Official PyTorch Implementation of the Longhorn Deep State Space Model
  ☆40Updated 3 months ago
• sustcsonglin / mamba-triton
  ☆45Updated 9 months ago
• maximzubkov / fft-scan
  Efficient PScan implementation in PyTorch
  ☆15Updated 10 months ago
• facebookresearch / bc-irl
  Implementation of BC-IRL and other IRL baselines
  ☆25Updated last year
• google-deepmind / spectral_ssm
  ☆28Updated 7 months ago
• brownirl / lambda_discrepancy
  Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
  ☆14Updated 3 weeks ago
• AlirezaMorsali / MLP-Attention
  ☆15Updated 10 months ago
• zdhNarsil / Distributional-GFlowNets
  Code for our TMLR paper "Distributional GFlowNets with Quantile Flows".
  ☆10Updated 9 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆45Updated 3 weeks ago
• ml-jku / L2M
  Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)
  ☆52Updated last month
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆46Updated last year
• OpenNLPLab / HGRN
  [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…
  ☆61Updated 6 months ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆29Updated 3 months ago