thomasahle / tensorgrad
Machine Learning with Symbolic Tensors
☆267Updated 2 months ago
Alternatives and similar repositories for tensorgrad:
Users that are interested in tensorgrad are comparing it to the libraries listed below
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆412Updated last week
- ☆431Updated 6 months ago
- Notebooks for the "JAX in Action" book☆147Updated 10 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆60Updated 9 months ago
- 🧱 Modula software package☆188Updated last month
- A pure NumPy implementation of Mamba.☆222Updated 10 months ago
- (WIP) A small but powerful, homemade PyTorch from scratch.☆550Updated 2 weeks ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆380Updated this week
- Compositional Linear Algebra☆475Updated last month
- ☆41Updated 4 months ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆92Updated 5 months ago
- ☆241Updated last year
- a categorical deep learning compiler☆201Updated 2 months ago
- Alex Krizhevsky's original code from Google Code☆191Updated 9 years ago
- ☆36Updated 4 months ago
- Puzzles for exploring transformers☆344Updated 2 years ago
- Interactive textbook on state-space models☆187Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆367Updated last month
- Multi-Threaded FP32 Matrix Multiplication on x86 CPUs☆349Updated 2 weeks ago
- Learnings and programs related to CUDA☆380Updated 2 months ago
- ☆217Updated 9 months ago
- For optimization algorithm research and development.☆512Updated this week
- Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach☆102Updated last year
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆174Updated 9 months ago
- Annotated version of the Mamba paper☆483Updated last year
- MIT IAP short course: Matrix Calculus for Machine Learning and Beyond☆479Updated 3 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆342Updated last month
- List of resources for learning Category Theory☆255Updated 5 months ago
- Simple Transformer in Jax☆136Updated 10 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆636Updated 2 months ago