joelburget / microjaxLinks
A tiny autograd engine with a Jax-like API
☆74Updated 2 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 5 months ago
- R.L. methods and techniques.☆200Updated 10 months ago
- Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …☆66Updated 2 months ago
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆206Updated last year
- Tensor library & inference framework for machine learning☆110Updated this week
- Heirarchical Navigable Small Worlds☆101Updated last month
- ☆34Updated 3 weeks ago
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆285Updated 2 weeks ago
- ☆248Updated last year
- ☆67Updated 4 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- ☆166Updated 2 months ago
- ☆196Updated 4 months ago
- Live-bending a foundation model’s output at neural network level.☆265Updated 5 months ago
- ☆47Updated 6 months ago
- Code for the Fractured Entangled Representation Hypothesis position paper!☆198Updated 4 months ago
- Designing bridge trusses with Pytorch autograd☆61Updated last year
- A star for organising blocks and playing with transformers.☆23Updated last year
- ☆231Updated 6 months ago
- Train neural networks that distill into logic circuits, using JAX☆62Updated 3 months ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆253Updated last year
- Run and explore Llama models locally with minimal dependencies on CPU☆189Updated 11 months ago
- Ever used asyncio and wished you hadn't? A tiny (~300 lines) event loop for Python.☆634Updated last month
- A playground to make it easy to try crazy things☆33Updated 3 months ago
- Algebraic enhancements for GEMM & AI accelerators☆280Updated 7 months ago
- A pure NumPy implementation of Mamba.☆223Updated last year
- Dead Simple LLM Abliteration☆232Updated 7 months ago
- time to learn mlx☆40Updated 2 weeks ago
- Grow virtual creatures in static and physics simulated environments.☆53Updated last year
- The Finite Field Assembly Programming Language☆36Updated 4 months ago