google-deepmind / treescopeLinks
An interactive HTML pretty-printer for machine learning research in IPython notebooks.
☆416Updated last month
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,780Updated last month
- ☆267Updated 10 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆374Updated 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…☆287Updated 5 months ago
- ☆246Updated 3 weeks ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆287Updated 9 months ago
- 🧱 Modula software package☆194Updated 2 months ago
- Compositional Linear Algebra☆474Updated this week
- A pure NumPy implementation of Mamba.☆223Updated 10 months ago
- Library for reading and processing ML training data.☆447Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆584Updated this week
- A simple & elegant experiment tracking framework that integrates persistence logic & best practices directly into Python☆523Updated 4 months ago
- Machine Learning with Symbolic Tensors☆277Updated 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…☆614Updated 2 months ago
- Uncertainty quantification with PyTorch☆356Updated last month
- Open weights language model from Google DeepMind, based on Griffin.☆639Updated last week
- ☆178Updated this week
- ☆431Updated 7 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆380Updated last month
- For optimization algorithm research and development.☆518Updated this week
- ☆150Updated 9 months ago
- TensorDict is a pytorch dedicated tensor container.☆925Updated this week
- An implementation of PSGD Kron second-order optimizer for PyTorch☆91Updated 2 months ago
- AutoBound automatically computes upper and lower bounds on functions.☆359Updated 11 months ago
- Named Tensors for Legible Deep Learning in JAX☆176Updated last month
- Orbax provides common checkpointing and persistence utilities for JAX users☆382Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,415Updated last month
- Official codebase for the paper "Beyond A* Better Planning with Transformers via Search Dynamics Bootstrapping".☆367Updated 11 months ago
- Next Generation Experimental Tracking for Machine Learning Operations☆238Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆276Updated 11 months ago