ezyang / stride-visualizer

Related projects ⓘ

Alternatives and complementary repositories for stride-visualizer

• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆129Updated 2 years ago
• mlcommons / training_results_v0.7
  This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.
  ☆56Updated last year
• mcarilli / mixed_precision_references
  Personal collection of references for high performance mixed precision training.
  ☆41Updated 5 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• mlcommons / training_results_v1.0
  This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.
  ☆37Updated 8 months ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 3 months ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆130Updated 2 months ago
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 2 years ago
• pytorch / multipy
  torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…
  ☆176Updated this week
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• MathInf / toroidal
  a lightweight transformer library for PyTorch
  ☆72Updated 3 years ago
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆63Updated 2 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆148Updated 3 years ago
• NVIDIA / dllogger
  A logging tool for deep learning.
  ☆52Updated 2 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆125Updated 3 years ago
• ezyang / pytorch-unattached
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆20Updated last year
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆39Updated 7 months ago
• albanD / subclass_zoo
  ☆148Updated 5 months ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• nv-legate / legate.pandas
  An Aspiring Drop-In Replacement for Pandas at Scale
  ☆74Updated 3 years ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 3 years ago
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆36Updated 2 months ago
• mlcommons / logging
  MLPerf™ logging library
  ☆30Updated this week
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆146Updated 2 weeks ago