thomasahle / tensorgradLinks
Machine Learning with Symbolic Tensors
☆299Updated 3 weeks ago
Alternatives and similar repositories for tensorgrad
Users that are interested in tensorgrad are comparing it to the libraries listed below
Sorting:
- ☆435Updated 8 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated 10 months ago
- Notebooks for the "Deep Learning with JAX" book☆149Updated 2 weeks ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆422Updated last month
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆399Updated 2 weeks ago
- Alex Krizhevsky's original code from Google Code☆192Updated 9 years ago
- A pure NumPy implementation of Mamba.☆223Updated 11 months ago
- For optimization algorithm research and development.☆521Updated this week
- Compositional Linear Algebra☆477Updated 3 weeks ago
- 🧱 Modula software package☆200Updated 2 months ago
- ☆270Updated 11 months ago
- Puzzles for exploring transformers☆350Updated 2 years ago
- Implementation of Diffusion Transformer (DiT) in JAX☆278Updated last year
- Getting crystal-like representations with harmonic loss☆190Updated 2 months ago
- Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach☆109Updated last year
- Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…☆649Updated last year
- Diffusion on syntax trees for program synthesis☆459Updated 11 months ago
- Multi-Threaded FP32 Matrix Multiplication on x86 CPUs☆352Updated 2 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆288Updated 9 months ago
- ☆248Updated last year
- Solve puzzles to improve your tinygrad skills!☆132Updated 3 months ago
- Open weights language model from Google DeepMind, based on Griffin.☆641Updated 2 weeks ago
- The boundary of neural network trainability is fractal☆207Updated last year
- ☆157Updated 2 months ago
- a categorical deep learning compiler☆202Updated 3 months ago
- Patched Attention for Nonlinear Dynamics☆147Updated 3 weeks ago
- (WIP) A small but powerful, homemade PyTorch from scratch.☆555Updated this week
- Deep Learning, an Energy Approach☆139Updated 2 weeks ago
- Annotated version of the Mamba paper☆485Updated last year