Alternatives and similar repositories for jasmine

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

• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆116Updated last year
• kvfrans / splus
  ☆122Updated 6 months ago
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆117Updated 2 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated last month
• kvfrans / lmpo
  ☆131Updated last month
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆206Updated last year
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated 2 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆109Updated 3 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆97Updated 11 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆29Updated last year
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆72Updated last year
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆36Updated 5 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated last year
• vladisai / PLDM
  ☆51Updated 2 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆120Updated last year
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆148Updated 8 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆46Updated 3 years ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆81Updated 10 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• danijar / elements
  Building blocks for productive research
  ☆67Updated this week
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 4 years ago
• Reytuag / transformerXL_PPO_JAX
  ☆89Updated last year
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 3 months ago
• PufferAI / PufferTank
  ☆59Updated 6 months ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆32Updated 2 years ago
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆65Updated 10 months ago
• DramaCow / jaxued
  ☆91Updated 4 months ago