joelburget / microjaxLinks
A tiny autograd engine with a Jax-like API
☆61Updated this week
Alternatives and similar repositories for microjax
Users that are interested in microjax are comparing it to the libraries listed below
Sorting:
- Framework for specifying and proving properties—such as robustness, fairness, and interpretability—of machine learning models using Lean …☆63Updated 4 months ago
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆126Updated 2 months ago
- DiscoGrad - automatically differentiate across conditional branches in C++ programs☆203Updated 10 months ago
- ☆34Updated 5 months ago
- R.L. methods and techniques.☆196Updated 7 months ago
- Designing bridge trusses with Pytorch autograd☆61Updated last year
- Code for the Fractured Entangled Representation Hypothesis position paper!☆111Updated last month
- Rewriting Principia Mathematica in Lean☆132Updated 7 months ago
- ☆64Updated last month
- ☆248Updated last year
- a categorical deep learning compiler☆203Updated 4 months ago
- Heirarchical Navigable Small Worlds☆97Updated 3 months ago
- Machine Learning with Symbolic Tensors☆313Updated last month
- Tensor library & inference framework for machine learning☆99Updated 2 weeks ago
- A star for organising blocks and playing with transformers.☆23Updated last year
- Live-bending a foundation model’s output at neural network level.☆262Updated 3 months ago
- time to learn mlx☆40Updated last month
- A playground to make it easy to try crazy things☆33Updated 3 weeks ago
- A pure NumPy implementation of Mamba.☆225Updated last year
- ☆47Updated 3 months ago
- Train neural networks that distill into logic circuits, using JAX☆61Updated last month
- High-Performance Klong array language in Python.☆303Updated last week
- Editor with LLM generation tree exploration☆71Updated 5 months ago
- ☆195Updated 2 months ago
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆285Updated last week
- ☆153Updated last week
- RLHF (Supervised fine-tuning, reward model, and PPO) step-by-step in 3 Jupyter notebooks☆111Updated 3 weeks ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆424Updated 2 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated 11 months ago
- ☆40Updated 2 months ago