Alternatives and similar repositories for jasmine

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

• kvfrans / splus
  ☆115Updated last month
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆105Updated 10 months ago
• PufferAI / PufferTank
  ☆42Updated 3 weeks ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆41Updated this week
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆130Updated last year
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆191Updated 11 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆140Updated 9 months ago
• kvfrans / lmpo
  ☆95Updated 3 weeks ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated last week
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆76Updated 7 months ago
• radarFudan / mamba-minimal-jax
  ☆31Updated 8 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆78Updated 5 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆32Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆34Updated 7 months ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆65Updated 7 months ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆33Updated 9 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆35Updated 9 months ago
• cloneofsimo / min-fsdp
  ☆83Updated last year
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆61Updated 5 months ago
• cloneofsimo / scaling-guide
  WIP
  ☆94Updated 11 months ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated last year
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆23Updated 2 months ago
• CLAIRE-Labo / EvoTune
  Efficiently discovering algorithms via LLMs with evolutionary search and reinforcement learning.
  ☆104Updated 3 weeks ago
• CLAIRE-Labo / flash_attention
  A basic pure pytorch implementation of flash attention
  ☆16Updated 9 months ago
• jennyzzt / omni
  OMNI: Open-endedness via Models of human Notions of Interestingness
  ☆52Updated 6 months ago
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆59Updated 5 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆67Updated 11 months ago