Alternatives and similar repositories for microjax

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

• valine / training-hot-swap
  Pytorch script hot swap: Change code without unloading your LLM from VRAM
  ☆126Updated 4 months ago
• ivanbelenky / RL
  R.L. methods and techniques.
  ☆199Updated 9 months ago
• fraware / leanverifier
  Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …
  ☆64Updated 3 weeks ago
• koogle / mlx-playground
  time to learn mlx
  ☆40Updated 2 months ago
• DiscoGrad / DiscoGrad
  DiscoGrad - automatically differentiate across conditional branches in C++ programs
  ☆204Updated 11 months ago
• mlecauchois / micrograd-cuda
  ☆248Updated last year
• nirw4nna / dsc
  Tensor library & inference framework for machine learning
  ☆107Updated this week
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆224Updated last year
• eschluntz / PytorchBridge
  Designing bridge trusses with Pytorch autograd
  ☆61Updated last year
• gvelesandro / constructor-theory-simulator
  ☆66Updated 2 months ago
• jmward01 / lmplay
  A playground to make it easy to try crazy things
  ☆33Updated 2 months ago
• trevorpogue / algebraic-nnhw
  Algebraic enhancements for GEMM & AI accelerators
  ☆278Updated 5 months ago
• em-llm / EM-LLM-model
  ☆220Updated 5 months ago
• utkuozbulak / pytorch-simple-diffusion
  ☆47Updated 4 months ago
• westoncb / nonlinear-optics-sandbox
  ☆34Updated 7 months ago
• joennlae / tensorli
  Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).
  ☆253Updated last year
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• sdiehl / sympy-mcp
  A MCP server for symbolic manipulation of mathematical expressions
  ☆38Updated last month
• babycommando / neuralgraffiti
  Live-bending a foundation model’s output at neural network level.
  ☆266Updated 4 months ago
• rentruewang / bocoel
  Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…
  ☆286Updated 3 weeks ago
• dicroce / hnsw
  Heirarchical Navigable Small Worlds
  ☆101Updated 2 weeks ago
• patrick-kidger / tinyio
  Ever used asyncio and wished you hadn't? A tiny (~300 lines) event loop for Python.
  ☆557Updated last week
• slashml / amd_inference
  Docker-based inference engine for AMD GPUs
  ☆231Updated 10 months ago
• blackhole89 / autopen
  Editor with LLM generation tree exploration
  ☆73Updated 6 months ago
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆203Updated 5 months ago
• thomasahle / tensorgrad
  Machine Learning with Symbolic Tensors
  ☆326Updated 2 months ago
• drubinstein / pokemonred_puffer
  ☆159Updated last month
• briangu / klongpy
  High-Performance Klong array language in Python.
  ☆307Updated last month
• Foreseerr / TScale
  ☆197Updated 3 months ago
• bclarkson-code / Tricycle
  Autograd to GPT-2 completely from scratch
  ☆116Updated 2 weeks ago