Alternatives and similar repositories for public_notebooks

Users that are interested in public_notebooks are comparing it to the libraries listed below

• gpu-mode / profiling-cuda-in-torch
  ☆178Updated last year
• dlsyscourse / hw0
  ☆54Updated 4 months ago
• nvixnu / pmpp__programming_massively_parallel_processors
  Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…
  ☆77Updated 5 years ago
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆27Updated last year
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆114Updated last year
• BoliboliWJY / Programming-Massively-Parallel-Processors-A-Hands-on-Approach-4th
  ☆14Updated 8 months ago
• HenryNdubuaku / cuda-tutorials
  CUDA tutorials for Maths & ML tutorials with examples, covers multi-gpus, fused attention, winograd convolution, reinforcement learning.
  ☆208Updated 7 months ago
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆289Updated 4 months ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆154Updated 2 years ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆163Updated last year
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆73Updated last year
• qhliu26 / Dive-into-Big-Model-Training
  📑 Dive into Big Model Training
  ☆116Updated 3 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated 2 years ago
• VikParuchuri / triton_tutorial
  Tutorials for Triton, a language for writing gpu kernels
  ☆72Updated 2 years ago
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆92Updated 2 years ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆175Updated 9 months ago
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆99Updated last year
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆46Updated last year
• stanford-cs336 / spring2024-lectures
  ☆409Updated last year
• sangmichaelxie / cs324_p2
  Project 2 (Building Large Language Models) for Stanford CS324: Understanding and Developing Large Language Models (Winter 2022)
  ☆105Updated 2 years ago
• cloneofsimo / min-fsdp
  ☆92Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated 2 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated last year
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆164Updated 2 weeks ago
• stanford-cs149 / asst1
  Stanford CS149 -- Assignment 1
  ☆145Updated 3 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆20Updated last year
• UmerHA / triton_util
  Make triton easier
  ☆50Updated last year
• tianlinxu312 / Everything-about-LLMs
  A work in progress. Trying to write about all interesting or necessary pieces in the current development of LLMs and generative AI. Gra…
  ☆199Updated 2 years ago
• stanford-cs149 / cs149gpt
  ☆79Updated 2 years ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆273Updated this week