joelburget / microjaxLinks
A tiny autograd engine with a Jax-like API
☆74Updated 6 months ago
Alternatives and similar repositories for microjax
Users that are interested in microjax are comparing it to the libraries listed below
Sorting:
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆125Updated 8 months ago
- R.L. methods and techniques.☆199Updated last week
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆209Updated last year
- time to learn mlx☆42Updated 4 months ago
- ☆250Updated last year
- Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …☆75Updated 5 months ago
- ☆47Updated 9 months ago
- Tensor library & inference framework for machine learning☆118Updated 3 months ago
- VortexNet: Neural Computing through Fluid Dynamics☆46Updated 11 months ago
- Proof of thought : LLM-based reasoning using Z3 theorem proving with multiple backend support (SMT2 and JSON DSL)☆363Updated 3 months ago
- ☆36Updated 4 months ago
- Heirarchical Navigable Small Worlds☆101Updated 5 months ago
- ☆69Updated 3 months ago
- A MCP server for symbolic manipulation of mathematical expressions☆48Updated 6 months ago
- Train neural networks that distill into logic circuits, using JAX☆64Updated 7 months ago
- Algebraic enhancements for GEMM & AI accelerators☆286Updated 10 months ago
- A character-level language diffusion model trained on Tiny Shakespeare☆830Updated this week
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆254Updated 2 years ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- High-Performance Klong array language in Python.☆311Updated last week
- Live-bending a foundation model’s output at neural network level.☆272Updated 9 months ago
- a categorical deep learning compiler☆207Updated 3 months ago
- Rewriting Principia Mathematica in Lean☆136Updated last week
- Evolution Pretraining Fully in Int Formats☆134Updated last month
- ☆176Updated last month
- A Full Transcript of the Lighthill Debate on AI from 1973, with Introductory Remarks☆33Updated last year
- Designing bridge trusses with Pytorch autograd☆61Updated last year
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆287Updated 4 months ago
- Build data processing and data analysis pipelines that leverage the power of LLMs 🧠☆246Updated last month
- Code for the Fractured Entangled Representation Hypothesis position paper!☆221Updated 2 months ago