Alternatives and similar repositories for jasmine

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

• kvfrans / lmpo
  ☆128Updated 3 weeks ago
• kvfrans / splus
  ☆121Updated 6 months ago
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆114Updated last year
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• RyanNavillus / Syllabus
  Synchronized Curriculum Learning for RL Agents
  ☆116Updated last month
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆132Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month
• ml-jku / LRAM
  A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks
  ☆36Updated 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
  ☆145Updated 7 months ago
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆201Updated last year
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆104Updated 2 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆97Updated 10 months ago
• martin-marek / batch-size
  📄Small Batch Size Training for Language Models
  ☆68Updated 2 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆71Updated 11 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆176Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated 11 months ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆81Updated 10 months ago
• cloneofsimo / min-fsdp
  ☆91Updated last year
• lucidrains / SAC-pytorch
  Implementation of Soft Actor Critic and some of its improvements in Pytorch
  ☆60Updated 10 months ago
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆207Updated last week
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 2 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆102Updated 2 months ago
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆75Updated 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
• vladisai / PLDM
  ☆50Updated last month
• google-deepmind / nanodo
  ☆285Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated last year