Alternatives and similar repositories for zephyr

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

• packquickly / schedule_free_optx
  Schedule free optimiser implemented in JAX using Optimistix
  ☆14Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆87Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆200Updated 2 months ago
• cgarciae / einop
  ☆60Updated 3 years ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆59Updated 6 months ago
• codingfisch / flashrl
  Fast reinforcement learning 💨
  ☆24Updated 3 months ago
• anh-tong / nanoGPT-equinox
  nanoGPT using Equinox
  ☆13Updated 2 years ago
• thebuckleylab / jpc
  Flexible Inference for Predictive Coding Networks in JAX.
  ☆48Updated 3 weeks ago
• Artur-Galstyan / statedict2pytree
  ☆43Updated 3 weeks ago
• kvfrans / splus
  ☆104Updated 2 weeks ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆34Updated 8 months ago
• okarthikb / state-space-models
  ☆27Updated 11 months ago
• idoh / fast_mamba.np
  A pure and fast NumPy implementation of Mamba with cache support.
  ☆17Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆93Updated last week
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆32Updated last year
• yixiaoer / einshard
  Einsum-like high-level array sharding API for JAX
  ☆35Updated 11 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆204Updated last month
• jfpuget / ARC-AGI-Challenge-2024
  ☆55Updated 7 months ago
• J-Rosser-UK / Torch2Jax-DeepSeek-R1-Distill-Qwen-1.5B
  Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.
  ☆20Updated 4 months ago
• jax-ml / jax-ai-stack
  ☆193Updated this week
• enajx / HyperNCA
  ☆39Updated 3 years ago
• google-deepmind / nanodo
  ☆270Updated 11 months ago
• patrick-kidger / wadler_lindig
  A Wadler--Lindig pretty printer for Python
  ☆39Updated this week
• GallagherCommaJack / modulax
  ☆17Updated 10 months ago
• jax-ml / jax-llm-examples
  ☆126Updated last month
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆57Updated 2 years ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated 11 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 4 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated last month
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆10Updated last week