Alternatives and similar repositories for jasmine

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

• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆38Updated last year
• kvfrans / splus
  ☆120Updated 4 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆113Updated last year
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆197Updated last year
• kvfrans / lmpo
  ☆117Updated last week
• radarFudan / mamba-minimal-jax
  ☆34Updated 11 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated 10 months ago
• FLAIROx / cultural-accumulation
  ☆14Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last year
• rail-berkeley / SUPE
  This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."
  ☆34Updated 3 months ago
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆114Updated 2 months ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆79Updated 8 months ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated last month
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆69Updated 10 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆84Updated 9 months ago
• CEC-Agent / CEC
  Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"
  ☆31Updated 2 years ago
• 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
  ☆141Updated 5 months ago
• danijar / elements
  Building blocks for productive research
  ☆62Updated 2 months ago
• google-deepmind / nao_top10
  ☆19Updated 2 years ago
• sdan / nanoEBM
  minimal Energy-based transformer
  ☆32Updated this week
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆63Updated 3 weeks ago
• danijar / ninjax
  General Modules for JAX
  ☆69Updated last month
• PufferAI / PufferTank
  ☆51Updated 3 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• vladisai / PLDM
  ☆42Updated 3 months ago
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆116Updated last year