google-deepmind / treescopeLinks
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
β422Updated last month
Alternatives and similar repositories for treescope
Users that are interested in treescope are comparing it to the libraries listed below
Sorting:
- π§± Modula software packageβ200Updated 2 months ago
- β270Updated 11 months ago
- 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 6 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.β1,790Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.β287Updated 9 months ago
- β435Updated 8 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorchβ91Updated 2 months ago
- Machine Learning with Symbolic Tensorsβ299Updated 3 weeks ago
- Compositional Linear Algebraβ476Updated 3 weeks ago
- A pure NumPy implementation of Mamba.β223Updated 11 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jaxβ595Updated this week
- A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and fullβ¦β615Updated 2 months ago
- β188Updated this week
- For optimization algorithm research and development.β521Updated this week
- Library for reading and processing ML training data.β457Updated this week
- R.L. methods and techniques.β190Updated 7 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.β382Updated 2 months ago
- β182Updated this week
- Pytorch script hot swap: Change code without unloading your LLM from VRAMβ126Updated 2 months ago
- β246Updated last month
- Orbax provides common checkpointing and persistence utilities for JAX usersβ390Updated this week
- A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Pythonβ524Updated 5 months ago
- TensorDict is a pytorch dedicated tensor container.β930Updated this week
- Named Tensors for Legible Deep Learning in JAXβ180Updated last month
- Uncertainty quantification with PyTorchβ361Updated 2 months ago
- Next Generation Experimental Tracking for Machine Learning Operationsβ315Updated 3 weeks ago
- β150Updated 10 months ago
- CLU lets you write beautiful training loops in JAX.β345Updated this week
- Puzzles for exploring transformersβ349Updated 2 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvementβ¦β385Updated this week