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.
☆288Updated 8 months ago
Alternatives and similar repositories for comgra
Users that are interested in comgra are comparing it to the libraries listed below
Sorting:
- A pure NumPy implementation of Mamba.☆224Updated last year
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆443Updated 3 weeks ago
- ☆144Updated 2 years ago
- Automatic gradient descent☆210Updated 2 years ago
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆349Updated last year
- Python library for designing and training your own Diffusion Models with PyTorch☆287Updated 2 months ago
- ☆150Updated last year
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆286Updated 3 weeks ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated last year
- Teaching transformers to play chess☆138Updated 7 months ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆390Updated last week
- ☆249Updated last year
- ☆307Updated last year
- Official implementation of MetaTree: Learning a Decision Tree Algorithm with Transformers☆113Updated 11 months ago
- git extension for {collaborative, communal, continual} model development☆216Updated 9 months ago
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆622Updated 5 months ago
- run paligemma in real time☆131Updated last year
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆120Updated last year
- Highly commented implementations of Transformers in PyTorch☆137Updated 2 years ago
- An interactive exploration of Transformer programming.☆269Updated last year
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆608Updated 5 months ago
- Memory mapped numpy arrays of varying shapes☆300Updated last year
- Pytorch script hot swap: Change code without unloading your LLM from VRAM☆126Updated 4 months ago
- Puzzles for exploring transformers☆367Updated 2 years ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆120Updated 6 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆400Updated 4 months ago
- 🤖 A PyTorch library of curated Transformer models and their composable components☆892Updated last year
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 8 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆373Updated last year
- Visualize the intermediate output of Mistral 7B☆368Updated 7 months ago