Alternatives and similar repositories for pydra

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

• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆166Updated last month
• google-deepmind / nanodo
  ☆277Updated last year
• anthropics / toy-models-of-superposition
  Notebooks accompanying Anthropic's "Toy Models of Superposition" paper
  ☆129Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated this week
• modula-systems / modula
  🧱 Modula software package
  ☆231Updated 2 weeks ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆59Updated 2 years ago
• thestephencasper / everything-you-need
  we got you bro
  ☆36Updated last year
• ayaka14732 / jax-smi
  JAX Synergistic Memory Inspector
  ☆179Updated last year
• stanford-crfm / levanter
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆648Updated last week
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆142Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆143Updated 10 months ago
• cloneofsimo / min-fsdp
  ☆87Updated last year
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆127Updated 8 months ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆208Updated last year
• anthropics / PySvelte
  A library for bridging Python and HTML/Javascript (via Svelte) for creating interactive visualizations
  ☆196Updated 3 years ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆132Updated 2 years ago
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆131Updated last year
• apartresearch / interpretability-starter
  🧠 Starter templates for doing interpretability research
  ☆73Updated 2 years ago
• r-three / git-theta
  git extension for {collaborative, communal, continual} model development
  ☆216Updated 9 months ago
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆103Updated 3 months ago
• Sea-Snell / JAX_llama
  Inference code for LLaMA models in JAX
  ☆118Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆201Updated last week
• jax-ml / jax-llm-examples
  Minimal yet performant LLM examples in pure JAX
  ☆151Updated this week
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆130Updated last month
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆156Updated 2 months ago
• TomFrederik / unseal
  Mechanistic Interpretability for Transformer Models
  ☆51Updated 3 years ago
• likenneth / othello_world
  Emergent world representations: Exploring a sequence model trained on a synthetic task
  ☆188Updated 2 years ago
• goodfire-ai / spd
  Stochastic Parameter Decomposition
  ☆37Updated this week
• moirage / alignment-research-dataset
  A dataset of alignment research and code to reproduce it
  ☆77Updated 2 years ago