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 8 months ago
• idoh / fast_mamba.np
  A pure and fast NumPy implementation of Mamba with cache support.
  ☆17Updated 8 months ago
• okarthikb / state-space-models
  ☆27Updated 7 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆29Updated 4 months ago
• nreHieW / r-nn
  Tensor library with autograd using only Rust's standard library
  ☆65Updated 7 months ago
• hyperevolnet / Terminator
  The official repository for HyperZ⋅Z⋅W Operator Connects Slow-Fast Networks for Full Context Interaction.
  ☆31Updated 2 weeks ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆47Updated 3 months ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆59Updated 2 months ago
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆76Updated 2 months ago
• swairshah / Intensify
  coloring terminal text with intensities (used for plotting probability, entropy with tokens)
  ☆12Updated 4 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆61Updated 6 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆59Updated 6 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025).
  ☆81Updated 3 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆146Updated this week
• jax-ml / jax-ai-stack
  ☆127Updated this week
• bloc97 / DeMo
  DeMo: Decoupled Momentum Optimization
  ☆181Updated 2 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆75Updated 4 months ago
• nikhilvyas / SOAP
  ☆159Updated 2 months ago
• misko / human_descent
  ☆36Updated 2 months ago
• Artur-Galstyan / statedict2pytree
  ☆40Updated 2 months ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆143Updated 7 months ago
• johnryan465 / pscan
  ☆37Updated last year
• Zyphra / Zamba2
  PyTorch implementation of models from the Zamba2 series.
  ☆176Updated 3 weeks ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆107Updated 3 weeks ago
• herrmann / rustorch
  "PyTorch in Rust"
  ☆16Updated last year
• shivance / picograd
  Rust Implementation of micrograd
  ☆51Updated 7 months ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated 3 weeks ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆136Updated 8 months ago