johnmarktaylor91 / torchlensLinks
Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
☆594Updated 3 months 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.☆929Updated last week
- 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 6 months ago
- Tensors, for human consumption☆1,257Updated this week
- Puzzles for exploring transformers☆350Updated 2 years ago
- ☆150Updated 10 months ago
- Interactively inspect module inputs, outputs, parameters, and gradients.☆341Updated last month
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆381Updated 2 months ago
- ☆436Updated 8 months ago
- ☆270Updated 11 months ago
- For optimization algorithm research and development.☆520Updated this week
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,796Updated this week
- torchview: visualize pytorch models☆955Updated last month
- UNet diffusion model in pure CUDA☆608Updated 11 months ago
- TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…☆117Updated 4 months ago
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆438Updated 9 months ago
- Compositional Linear Algebra☆477Updated last month
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆426Updated 6 months ago
- Scalable and Performant Data Loading☆278Updated this week
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆424Updated last month
- Annotated version of the Mamba paper☆485Updated last year
- What would you do with 1000 H100s...☆1,055Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆801Updated 2 weeks ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆252Updated 3 months ago
- Puzzles for learning Triton☆1,726Updated 7 months ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,431Updated last month
- ☆65Updated last year
- Schedule-Free Optimization in PyTorch☆2,180Updated last month
- A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.☆909Updated last year
- ☆778Updated 3 weeks ago