Alternatives and similar repositories for jaxtransformer

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

• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆21Updated 5 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆41Updated last year
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆22Updated 2 years ago
• sukhijab / maxinforl_jax
  ☆27Updated last month
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆28Updated last year
• kvfrans / splus
  ☆124Updated 8 months ago
• haoliuhl / tux
  Tools and Utils for Experiments (TUX)
  ☆15Updated last year
• StoneT2000 / rl-robotics-speedrun
  speed-running solving robot manipulation tasks
  ☆24Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆61Updated 8 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆16Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆115Updated last year
• lili-chen / SEER
  Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.
  ☆21Updated 4 years ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆129Updated 3 weeks ago
• dunnolab / laom
  Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025
  ☆33Updated 7 months ago
• notmahi / disk
  PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…
  ☆20Updated 3 years ago
• marc-rigter / waker
  Official code for "Reward-Free Curricula for Training Robust World Models", ICLR 2024.
  ☆28Updated 2 years ago
• yuqingd / cusp
  ☆15Updated 3 years ago
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆22Updated last year
• frt03 / jax_dt
  Minimal Decision Transformer Implementation written in Jax (Flax).
  ☆17Updated 3 years ago
• facebookresearch / entity-factored-rl
  Source code for the paper "Policy Architectures for Compositional Generalization in Control"
  ☆30Updated 3 years ago
• Snagnar / Hieros
  Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper
  ☆21Updated last year
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆36Updated 2 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆46Updated last year
• robfiras / s2pg
  Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"
  ☆25Updated last year
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆41Updated 6 months ago
• si0wang / COPlanner
  ☆23Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆36Updated 4 years ago
• mohmdelsayed / HesScale
  Scalable Computation of Hessian Diagonals
  ☆14Updated last year