google-deepmind / treescopeLinks
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
☆426Updated 3 months ago
Alternatives and similar repositories for treescope
Users that are interested in treescope are comparing it to the libraries listed below
Sorting:
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,805Updated last month
- 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…☆288Updated 7 months ago
- ☆274Updated last year
- 🧱 Modula software package☆210Updated last week
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆291Updated 11 months ago
- Next Generation Experimental Tracking for Machine Learning Operations☆333Updated 2 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆388Updated 3 months ago
- ☆443Updated 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…☆620Updated 4 months ago
- Compositional Linear Algebra☆483Updated 2 months ago
- A pure NumPy implementation of Mamba.☆223Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆627Updated this week
- Uncertainty quantification with PyTorch☆366Updated 3 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆445Updated this week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆94Updated last week
- ☆247Updated last month
- Open weights language model from Google DeepMind, based on Griffin.☆645Updated 2 months ago
- ☆150Updated 11 months ago
- Ever used asyncio and wished you hadn't? A tiny (~200 lines) event loop for Python.☆435Updated this week
- A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python☆525Updated 6 months ago
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated last year
- Machine Learning with Symbolic Tensors☆324Updated 2 months ago
- Named Tensors for Legible Deep Learning in JAX☆197Updated this week
- ☆186Updated this week
- For optimization algorithm research and development.☆525Updated this week
- A plotting tool that outputs Line Rider maps, so you can watch a man on a sled scoot down your loss curves. 🎿☆330Updated 11 months ago
- Manipulating Python Programs☆677Updated last week
- Library for reading and processing ML training data.☆484Updated last week
- Scalable and Performant Data Loading☆290Updated last week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆389Updated this week