johnmarktaylor91 / torchlensLinks
Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
☆616Updated 2 weeks ago
Alternatives and similar repositories for torchlens
Users that are interested in torchlens are comparing it to the libraries listed below
Sorting:
- TensorDict is a pytorch dedicated tensor container.☆970Updated this week
- Tensors, for human consumption☆1,312Updated this week
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆120Updated 7 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…☆290Updated 10 months ago
- Interactively inspect module inputs, outputs, parameters, and gradients.☆352Updated 4 months ago
- Puzzles for exploring transformers☆370Updated 2 years ago
- ☆150Updated last year
- torchview: visualize pytorch models☆991Updated 4 months ago
- For optimization algorithm research and development.☆539Updated 2 weeks ago
- Annotated version of the Mamba paper☆489Updated last year
- ☆455Updated 11 months ago
- Exca - Execution and caching tool for python☆106Updated last week
- Universal Notation for Tensor Operations in Python.☆433Updated 6 months ago
- ☆785Updated last month
- Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"☆562Updated last year
- ☆282Updated last year
- Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…☆271Updated this week
- ViT Prisma is a mechanistic interpretability library for Vision and Video Transformers (ViTs).☆309Updated 2 months ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆303Updated 2 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆667Updated this week
- ☆508Updated 2 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Best practices & guides on how to write distributed pytorch training code☆494Updated this week
- Efficient optimizers☆265Updated last week
- Implementation of Diffusion Transformer (DiT) in JAX☆292Updated last year
- Everything you want to know about Google Cloud TPU☆546Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆293Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,823Updated 3 months ago
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆816Updated 2 months ago
- A repository for log-time feedforward networks☆222Updated last year