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
  ☆105Updated 11 months ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆131Updated last year
• kvfrans / splus
  ☆115Updated 2 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated last month
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆43Updated last week
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆34Updated last year
• radarFudan / mamba-minimal-jax
  ☆31Updated 9 months ago
• PufferAI / PufferTank
  ☆44Updated last month
• kvfrans / lmpo
  ☆101Updated last week
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆111Updated this week
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆34Updated 7 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆143Updated 9 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆194Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated this week
• cloneofsimo / min-fsdp
  ☆87Updated last year
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆67Updated 8 months ago
• 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 6 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆76Updated 8 months ago
• conglu1997 / intelligent-go-explore
  Intelligent Go-Explore: Standing on the Shoulders of Giant Foundation Models
  ☆63Updated 6 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated last year
• jennyzzt / omni
  OMNI: Open-endedness via Models of human Notions of Interestingness
  ☆55Updated 7 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆35Updated 2 weeks ago
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆35Updated 9 months ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆93Updated 5 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆87Updated last year
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆23Updated 2 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year