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 5 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.☆419Updated last month
- A pure NumPy implementation of Mamba.☆223Updated 10 months ago
- ☆150Updated 9 months ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆287Updated 9 months ago
- Scalable and Performant Data Loading☆269Updated this week
- ☆301Updated 11 months ago
- TensorDict is a pytorch dedicated tensor container.☆925Updated this week
- Puzzles for exploring transformers☆347Updated 2 years ago
- Automatic gradient descent☆207Updated last year
- ☆267Updated 10 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆374Updated last month
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆586Updated this week
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆588Updated 2 months ago
- For optimization algorithm research and development.☆518Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆381Updated this week
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆116Updated 3 months ago
- Annotated version of the Mamba paper☆482Updated last year
- ☆431Updated 7 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
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,784Updated last month
- Memory mapped numpy arrays of varying shapes☆298Updated 11 months ago
- Uncertainty quantification with PyTorch☆357Updated last month
- Website for hosting the Open Foundation Models Cheat Sheet.☆267Updated 3 weeks ago
- ☆143Updated 2 years ago
- Teaching transformers to play chess☆125Updated 4 months ago
- Bayesian Optimization as a Coverage Tool for Evaluating LLMs. Accurate evaluation (benchmarking) that's 10 times faster with just a few l…☆284Updated last week
- Compositional Linear Algebra☆474Updated last week
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated 11 months ago
- Absolute minimalistic implementation of a GPT-like transformer using only numpy (<650 lines).☆251Updated last year
- ☆246Updated 3 weeks ago