sachinhosmani / torchvistaLinks
Interactive Pytorch forward pass visualization in notebooks
☆76Updated this week
Alternatives and similar repositories for torchvista
Users that are interested in torchvista are comparing it to the libraries listed below
Sorting:
- ☆103Updated 9 months ago
- ☆150Updated 9 months ago
- 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
- Implementation snake game based on Diffusion model☆92Updated 4 months ago
- An implementation of PSGD Kron second-order optimizer for PyTorch☆91Updated 2 months ago
- Highly commented implementations of Transformers in PyTorch☆136Updated last year
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆423Updated 5 months ago
- Getting crystal-like representations with harmonic loss☆187Updated 2 months ago
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆115Updated last year
- Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.…☆91Updated 4 months ago
- 🧱 Modula software package☆194Updated 2 months ago
- Efficient optimizers☆208Updated this week
- Exca - Execution and caching tool for python☆85Updated this week
- Best practices & guides on how to write distributed pytorch training code☆433Updated 3 months ago
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆35Updated 5 months ago
- ☆230Updated this week
- Software design principles for machine learning applications☆359Updated 2 months ago
- Train an adapter for any embedding model in under a minute☆101Updated last month
- Minimal example scripts of the Hugging Face Trainer, focused on staying under 150 lines☆197Updated last year
- A pure NumPy implementation of Mamba.☆223Updated 10 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
- An extension of the nanoGPT repository for training small MOE models.☆147Updated 2 months ago
- SETOL: SemiEmpirical Theory of (Deep) Learning☆16Updated last week
- The AdEMAMix Optimizer: Better, Faster, Older.☆183Updated 8 months ago
- Resources from the EleutherAI Math Reading Group☆53Updated 3 months ago
- An interactive HTML pretty-printer for machine learning research in IPython notebooks.☆419Updated last month
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆128Updated 3 weeks ago
- Machine Learning with Symbolic Tensors☆278Updated last week
- Actually Robust Training - Tool Inspired by Andrej Karpathy "Recipe for training neural networks". It allows you to decompose your Deep…☆45Updated last year
- (WIP) A small but powerful, homemade PyTorch from scratch.☆553Updated this week