FlorianDietz / comgraLinks
A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different angles at the click of a button.
☆296Updated last year
Alternatives and similar repositories for comgra
Users that are interested in comgra are comparing it to the libraries listed below
Sorting:
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆454Updated 4 months ago
- A pure NumPy implementation of Mamba.☆222Updated last year
- Automatic gradient descent☆215Updated 2 years ago
- ☆144Updated 2 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆297Updated last year
- ☆314Updated last year
- run paligemma in real time☆133Updated last year
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆355Updated last year
- ☆212Updated last year
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆122Updated last year
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆123Updated 10 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆404Updated this week
- 🤖 A PyTorch library of curated Transformer models and their composable components☆894Updated last year
- Open weights language model from Google DeepMind, based on Griffin.☆656Updated 6 months ago
- Helpers and such for working with Lambda Cloud☆51Updated 2 years ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆97Updated 4 months ago
- a small code base for training large models☆315Updated 7 months ago
- A repository for log-time feedforward networks☆223Updated last year
- For optimization algorithm research and development.☆553Updated this week
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆628Updated 9 months ago
- Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines☆196Updated last year
- Highly commented implementations of Transformers in PyTorch☆139Updated 2 years ago
- The repository for the code of the UltraFastBERT paper☆520Updated last year
- Teaching transformers to play chess☆144Updated last week
- Language Modeling with the H3 State Space Model☆521Updated 2 years ago
- Visualizing the internal board state of a GPT trained on chess PGN strings, and performing interventions on its internal board state and …☆219Updated last year
- Puzzles for exploring transformers☆380Updated 2 years ago
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆125Updated 8 months ago
- Python library for designing and training your own Diffusion Models with PyTorch☆290Updated 6 months ago
- Official implementation of MetaTree: Learning a Decision Tree Algorithm with Transformers☆115Updated last year