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.
☆287Updated 6 months ago
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.☆422Updated last month
- A pure NumPy implementation of Mamba.☆223Updated 11 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
- For optimization algorithm research and development.☆521Updated this week
- ☆270Updated 11 months ago
- Automatic gradient descent☆208Updated last year
- Puzzles for exploring transformers☆350Updated 2 years ago
- Open weights language model from Google DeepMind, based on Griffin.☆641Updated 3 weeks ago
- ☆303Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆385Updated this week
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆345Updated 10 months ago
- ☆435Updated 8 months ago
- ☆150Updated 10 months ago
- ☆143Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆94Updated 6 months ago
- ☆246Updated last month
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆252Updated last year
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆615Updated 3 months ago
- Annotated version of the Mamba paper☆485Updated last year
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆117Updated 4 months ago
- Highly commented implementations of Transformers in PyTorch☆136Updated last year
- 🤖 A PyTorch library of curated Transformer models and their composable components☆891Updated last year
- Visualizing the internal board state of a GPT trained on chess PGN strings, and performing interventions on its internal board state and …☆206Updated 7 months ago
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 9 months ago
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆592Updated 3 months ago
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆285Updated 3 weeks ago
- A repository for log-time feedforward networks☆222Updated last year
- An interactive exploration of Transformer programming.☆264Updated last year