mtbrandy / pytorch

Related projects ⓘ

Alternatives and complementary repositories for pytorch

• IBM / pytorch-large-model-support
  Large Model Support in PyTorch
  ☆132Updated 2 years ago
• noahgolmant / pytorch-lars
  "Layer-wise Adaptive Rate Scaling" in PyTorch
  ☆86Updated 3 years ago
• sara-nl / attention-sampling-pytorch
  This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".
  ☆40Updated last year
• NVlabs / AdaBatch
  AdaBatch: Adaptive Batch Sizes for Training Deep Neural Networks
  ☆41Updated 6 years ago
• eisen-ai / eisen-core
  Core functionality of Eisen
  ☆41Updated 4 years ago
• IBM / tensorflow-large-model-support
  Large Model Support in Tensorflow
  ☆201Updated 4 years ago
• DIAGNijmegen / StreamingCNN
  To train deep convolutional neural networks, the input data and the activations need to be kept in memory. Given the limited memory avail…
  ☆98Updated last year
• adityaiitb / pyprof2
  PyProf2: PyTorch Profiling tool
  ☆83Updated 4 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• topal-team / rockmate
  ☆33Updated last year
• csrhddlam / pytorch-checkpoint
  ☆165Updated 5 years ago
• gpuopenanalytics / pynvml
  Provide Python access to the NVML library for GPU diagnostics
  ☆220Updated 3 months ago
• ryujaehun / pytorch-gpu-benchmark
  Using the famous cnn model in Pytorch, we run benchmarks on various gpu.
  ☆227Updated 4 months ago
• dholt / slurm-gpu
  Scheduling GPU cluster workloads with Slurm
  ☆73Updated 6 years ago
• zasdfgbnm / autonvtx
  Automatically insert nvtx ranges to PyTorch models
  ☆17Updated 3 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆148Updated 3 years ago
• NVIDIA / nvtx-plugins
  Python bindings for NVTX
  ☆66Updated last year
• FabianFalck / unet-design
  Official PyTorch implementation of "A Unified Framework for U-Net Design and Analysis" (NeurIPS 2023).
  ☆13Updated last year
• evcu / pytorchpruner
  ☆22Updated 6 years ago
• tbung / pytorch-revnet
  Implementation of the reversible residual network in pytorch
  ☆101Updated 2 years ago
• jansel / pytorch-jit-paritybench
  ☆36Updated 5 months ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆116Updated 3 months ago
• mlcommons / training_results_v0.5
  This repository contains the results and code for the MLPerf™ Training v0.5 benchmark.
  ☆35Updated last year
• intel / torch-ccl
  oneCCL Bindings for Pytorch*
  ☆86Updated 2 weeks ago
• victorcampos7 / weightnorm-init
  Code for "How to Initialize your Network? Robust Initialization for WeightNorm & ResNets"
  ☆14Updated last year
• ezyang / nvprof2json
  Convert nvprof profiles into about:tracing compatible JSON files
  ☆67Updated 3 years ago
• ptillet / triton-llvm-releases
  ☆20Updated last year
• BradMcDanel / sdgp
  ☆11Updated 2 years ago
• waitwaitforget / modelparallel_pytorch
  Model Parallelism for pytorch training multiple network on multiple GPUs.
  ☆28Updated 6 years ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆493Updated 3 years ago