johnmarktaylor91 / torchlensLinks
Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
☆588Updated 3 months ago
Alternatives and similar repositories for torchlens
Users that are interested in torchlens are comparing it to the libraries listed below
Sorting:
- Interactively inspect module inputs, outputs, parameters, and gradients.☆338Updated 3 weeks 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…☆287Updated 5 months ago
- torchview: visualize pytorch models☆940Updated 2 weeks ago
- For optimization algorithm research and development.☆518Updated this week
- Puzzles for exploring transformers☆348Updated 2 years ago
- Annotated version of the Mamba paper☆482Updated last year
- An interactive exploration of Transformer programming.☆264Updated last year
- ☆433Updated 7 months ago
- ☆150Updated 9 months ago
- Tensors, for human consumption☆1,252Updated last week
- TensorDict is a pytorch dedicated tensor container.☆927Updated this week
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆586Updated last week
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆239Updated 3 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆380Updated last month
- UNet diffusion model in pure CUDA☆606Updated 11 months ago
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,422Updated last month
- Torch nn vizualization☆51Updated last year
- Implementation of https://srush.github.io/annotated-s4☆497Updated 2 years ago
- Install PyTorch distributions with computation backend auto-detection☆253Updated last month
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆546Updated this week
- Best practices & guides on how to write distributed pytorch training code☆433Updated 3 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆116Updated 3 months ago
- A repository for log-time feedforward networks☆222Updated last year
- ☆776Updated this week
- ☆267Updated 10 months ago
- Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"☆554Updated 11 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,784Updated last month
- Schedule-Free Optimization in PyTorch☆2,162Updated 2 weeks ago
- Building blocks for foundation models.☆502Updated last year