thomasahle / tensorgrad
Machine Learning with Symbolic Tensors
☆262Updated 2 weeks ago
Alternatives and similar repositories for tensorgrad:
Users that are interested in tensorgrad are comparing it to the libraries listed below
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 7 months ago
- Notebooks for the "JAX in Action" book☆133Updated 9 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆398Updated last week
- ☆423Updated 5 months ago
- a categorical deep learning compiler☆199Updated 3 weeks ago
- 🧱 Modula software package☆173Updated 2 weeks ago
- Compositional Linear Algebra☆465Updated last week
- (WIP) A small but powerful, homemade PyTorch from scratch.☆537Updated this week
- ☆242Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,746Updated 3 months ago
- The boundary of neural network trainability is fractal☆195Updated last year
- ☆214Updated 8 months ago
- For optimization algorithm research and development.☆498Updated this week
- A pure NumPy implementation of Mamba.☆219Updated 8 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆370Updated this week
- ☆41Updated 2 months ago
- Uncertainty quantification with PyTorch☆348Updated this week
- Implementation of Diffusion Transformer (DiT) in JAX☆269Updated 9 months ago
- Named Tensors for Legible Deep Learning in JAX☆167Updated this week
- Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach☆92Updated last year
- Solve puzzles. Learn CUDA.☆63Updated last year
- Multi-Threaded FP32 Matrix Multiplication on x86 CPUs☆341Updated last month
- Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…☆646Updated 11 months ago
- Gradient descent is cool and all, but what if we could delete it?☆103Updated 2 weeks ago
- ☆36Updated 3 months ago
- Simple Transformer in Jax☆136Updated 9 months ago
- Interactive textbook on state-space models☆184Updated last year
- Solve puzzles to improve your tinygrad skills!☆120Updated last week
- Alex Krizhevsky's original code from Google Code☆190Updated 9 years ago