zasdfgbnm / autonvtx

Related projects ⓘ

Alternatives and complementary repositories for autonvtx

• csarofeen / pytorch
  Tensors and Dynamic neural networks in Python with strong GPU acceleration
  ☆26Updated last year
• bwasti / pytorch_compiler_tutorial
  Codebase associated with the PyTorch compiler tutorial
  ☆44Updated 5 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months 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
• Quansight / pytest-pytorch
  pytest plugin for a better developer experience when working with the PyTorch test suite
  ☆43Updated 2 years ago
• nod-ai / transformer-benchmarks
  benchmarking some transformer deployments
  ☆26Updated last year
• zdevito / custom_loader
  ☆12Updated 3 years ago
• nunoplopes / torchy
  A tracing JIT compiler for PyTorch
  ☆12Updated 2 years ago
• VoVAllen / tf-dlpack
  DLPack for Tensorflow
  ☆36Updated 4 years ago
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆127Updated 2 months ago
• manishucsd / py-codegen
  ☆14Updated last month
• ray-project / ray_shuffling_data_loader
  A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…
  ☆18Updated last year
• 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…
  ☆174Updated this week
• fidelity / stoke
  A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…
  ☆66Updated last year
• octoml / synr
  A library for syntactically rewriting Python programs, pronounced (sinner).
  ☆70Updated 2 years ago
• NVIDIA / dllogger
  A logging tool for deep learning.
  ☆51Updated 2 years ago
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆86Updated last week
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• gmarkall / life-of-a-numba-kernel
  Worked example of the process from Python source to CUDA kernel execution with Numba
  ☆36Updated last month
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago
• NVIDIA / nvtx-plugins
  Python bindings for NVTX
  ☆66Updated last year
• ray-project / distml
  Distributed ML Optimizer
  ☆31Updated 3 years ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• ptillet / triton-llvm-releases
  ☆20Updated last year
• mlcommons / training_results_v0.7
  This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.
  ☆56Updated last year
• FrancescoSaverioZuppichini / pytorch-2.0-benchmark
  Benchmarking PyTorch 2.0 different models
  ☆21Updated last year
• facebookresearch / loop_nest
  Loop Nest - Linear algebra compiler and code generator.
  ☆22Updated 2 years ago
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 4 years ago