sachinhosmani / torchvistaLinks
Interactive Pytorch forward pass visualization in notebooks
☆673Updated last week
Alternatives and similar repositories for torchvista
Users that are interested in torchvista are comparing it to the libraries listed below
Sorting:
- Interactively inspect module inputs, outputs, parameters, and gradients.☆354Updated last month
- This repository provides a Python script to fetch and summarize research papers from arXiv using the free Gemini API☆257Updated 10 months ago
- Model Activity Visualiser☆520Updated 9 months ago
- Official repository of my book "A Hands-On Guide to Fine-Tuning LLMs with PyTorch and Hugging Face"☆773Updated 3 months ago
- A minimal PyTorch re-implementation of Qwen3 VL with a fancy CLI☆313Updated last month
- Learning Deep Representations of Data Distributions☆748Updated this week
- Muon is an optimizer for hidden layers in neural networks☆2,231Updated last week
- Implementation of Stable Diffusion with PyTorch☆360Updated 11 months ago
- A comprehensive book on neural networks and large language models in NLP☆552Updated last month
- Implementation of the sparse attention pattern proposed by the Deepseek team in their "Native Sparse Attention" paper☆793Updated 5 months ago
- When it comes to optimizers, it's always better to be safe than sorry☆399Updated 4 months ago
- Minimal and annotated implementations of key ideas from modern deep learning research.☆1,222Updated 4 months ago
- CUDA tutorials for Maths & ML tutorials with examples, covers multi-gpus, fused attention, winograd convolution, reinforcement learning.☆208Updated 7 months ago
- A straightforward method for training your LLM, from downloading data to generating text.☆507Updated 5 months ago
- [NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (T6) (https://arxiv.org/abs/2501.06425)☆445Updated this week
- Coding a Multimodal (Vision) Language Model from scratch in PyTorch with full explanation: https://www.youtube.com/watch?v=vAmKB7iPkWw☆588Updated last year
- Contains the public resources of Hands on GenAI book☆229Updated last year
- Static suckless single batch CUDA-only qwen3-0.6B mini inference engine☆542Updated 4 months ago
- A fully functional and simple Machine Learning library made entirely from scratch with Python.☆438Updated last month
- Labs for MIT 6.S184/6.S975, IAP 2025/2026☆301Updated last week
- First-principle implementations of groundbreaking AI algorithms using a wide range of deep learning frameworks, accompanied by supporting…☆181Updated 6 months ago
- Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computation (NeurIPS 2025)☆538Updated 4 months ago
- About This repository is a curated collection of the most exciting and influential CVPR 2025 papers. 🔥 [Paper + Code + Demo]☆840Updated 7 months ago
- Based on Nano-vLLM, a simple replication of vLLM with self-contained paged attention and flash attention implementation☆243Updated last week
- LeetGPU Challenges☆602Updated this week
- dLLM: Simple Diffusion Language Modeling☆1,633Updated 3 weeks ago
- Textbook on reinforcement learning from human feedback☆1,426Updated this week
- Implementation of all RL algorithms in a simpler way☆1,383Updated 5 months ago
- ☆411Updated 9 months ago
- From scratch implementation of a vision language model in pure PyTorch☆253Updated last year