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
  ☆124Updated 6 months ago
• fraware / leanverifier
  Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …
  ☆69Updated 3 months ago
• ivanbelenky / RL
  R.L. methods and techniques.
  ☆199Updated 11 months ago
• DiscoGrad / DiscoGrad
  DiscoGrad - automatically differentiate across conditional branches in C++ programs
  ☆208Updated last year
• nirw4nna / dsc
  Tensor library & inference framework for machine learning
  ☆113Updated last month
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• westoncb / nonlinear-optics-sandbox
  ☆36Updated 2 months ago
• DebarghaG / proofofthought
  Proof of thought : LLM-based reasoning using Z3 theorem proving with multiple backend support (SMT2 and JSON DSL)
  ☆350Updated 3 weeks ago
• Dicklesworthstone / the_lighthill_debate_on_ai
  A Full Transcript of the Lighthill Debate on AI from 1973, with Introductory Remarks
  ☆33Updated last year
• statusfailed / catgrad
  a categorical deep learning compiler
  ☆204Updated last month
• jmward01 / lmplay
  A playground to make it easy to try crazy things
  ☆33Updated last month
• dicroce / hnsw
  Heirarchical Navigable Small Worlds
  ☆101Updated 3 months ago
• mlecauchois / micrograd-cuda
  ☆248Updated last year
• briangu / klongpy
  High-Performance Klong array language in Python.
  ☆311Updated last month
• slightknack / difflogic
  Train neural networks that distill into logic circuits, using JAX
  ☆63Updated 5 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…
  ☆285Updated 2 months ago
• babycommando / neuralgraffiti
  Live-bending a foundation model’s output at neural network level.
  ☆270Updated 7 months ago
• samim23 / vortexnet
  VortexNet: Neural Computing through Fluid Dynamics
  ☆46Updated 9 months ago
• gvelesandro / constructor-theory-simulator
  ☆68Updated 3 weeks ago
• akarshkumar0101 / fer
  Code for the Fractured Entangled Representation Hypothesis position paper!
  ☆204Updated last week
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆70Updated 8 months ago
• koogle / mlx-playground
  time to learn mlx
  ☆40Updated last month
• idoh / mamba.np
  A pure NumPy implementation of Mamba.
  ☆223Updated last year
• sdiehl / sympy-mcp
  A MCP server for symbolic manipulation of mathematical expressions
  ☆42Updated 4 months ago
• anishathalye / semlib
  Build data processing and data analysis pipelines that leverage the power of LLMs 🧠
  ☆226Updated 3 weeks ago
• ndrwnaguib / principia
  Rewriting Principia Mathematica in Lean
  ☆135Updated 2 months ago
• utkuozbulak / pytorch-simple-diffusion
  ☆47Updated 7 months ago
• drubinstein / pokemonred_puffer
  ☆172Updated 4 months ago
• eschluntz / PytorchBridge
  Designing bridge trusses with Pytorch autograd
  ☆61Updated last year
• rayking99 / BlockStar
  A star for organising blocks and playing with transformers.
  ☆23Updated last year