google-deepmind / treescopeLinks
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
☆425Updated 2 months ago
Alternatives and similar repositories for treescope
Users that are interested in treescope are comparing it to the libraries listed below
Sorting:
- 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…☆287Updated 7 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,797Updated 3 weeks ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 10 months ago
- ☆273Updated 11 months ago
- 🧱 Modula software package☆202Updated 3 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆385Updated 3 months ago
- ☆246Updated last week
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…☆618Updated 3 months ago
- A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python☆524Updated 5 months ago
- ☆440Updated 8 months ago
- Machine Learning with Symbolic Tensors☆313Updated last month
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated last year
- A pure NumPy implementation of Mamba.☆225Updated last year
- Compositional Linear Algebra☆478Updated last month
- Next Generation Experimental Tracking for Machine Learning Operations☆321Updated last month
- Manipulating Python Programs☆665Updated last month
- Named Tensors for Legible Deep Learning in JAX☆185Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆607Updated this week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆92Updated 3 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆424Updated this week
- Uncertainty quantification with PyTorch☆362Updated 2 months ago
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆369Updated last year
- For optimization algorithm research and development.☆521Updated this week
- Open weights language model from Google DeepMind, based on Griffin.☆644Updated last month
- Puzzles for exploring transformers☆354Updated 2 years ago
- Library for reading and processing ML training data.☆474Updated this week
- Cost aware hyperparameter tuning algorithm☆162Updated last year
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆386Updated last week
- ☆186Updated 2 weeks ago
- The history files when recording human interaction while solving ARC tasks☆113Updated this week