wkentaro / pytorch-for-numpy-users

Related projects: ⓘ

• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆539Updated 3 years ago
• pytorch / tnt
  A lightweight library for PyTorch training tools and utilities
  ☆1,658Updated this week
• kevinzakka / pytorch-goodies
  PyTorch Boilerplate For Research
  ☆603Updated 2 years ago
• nitrain / nitrain
  Train AI models efficiently on medical images using any framework
  ☆1,860Updated 3 months ago
• pytorchbearer / torchbearer
  torchbearer: A model fitting library for PyTorch
  ☆633Updated 9 months ago
• chrischoy / pytorch-custom-cuda-tutorial
  Tutorial for building a custom CUDA function for Pytorch
  ☆515Updated 5 years ago
• Spandan-Madan / A-Collection-of-important-tasks-in-pytorch
  Everyday things people use in Pytorch. No need to spend hours reading Pytorch forums trying to find them!
  ☆279Updated 2 years ago
• zasdfgbnm / TorchSnooper
  Debug PyTorch code using PySnooper
  ☆800Updated 3 years ago
• kevinzakka / hypersearch
  ☆399Updated this week
• prigoyal / pytorch_memonger
  Experimental ground for optimizing memory of pytorch models
  ☆356Updated 6 years ago
• waleedka / hiddenlayer
  Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
  ☆1,789Updated 7 months ago
• davidcpage / cifar10-fast
  ☆531Updated 2 years ago
• timgaripov / swa
  Stochastic Weight Averaging in PyTorch
  ☆959Updated 3 years ago
• awwong1 / torchprof
  PyTorch layer-by-layer model profiler
  ☆608Updated 3 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,013Updated last month
• szagoruyko / pytorchviz
  A small package to create visualizations of PyTorch execution graphs
  ☆3,170Updated 5 months ago
• pytorch / contrib
  Implementations of ideas from recent papers
  ☆391Updated 3 years ago
• jacobkimmel / pytorch_modelsize
  Estimates the size of a PyTorch model in memory
  ☆351Updated 4 years ago
• FrancescoSaverioZuppichini / Pytorch-how-and-when-to-use-Module-Sequential-ModuleList-and-ModuleDict
  Code for my medium article
  ☆357Updated 4 years ago
• williamFalcon / test-tube
  Python library to easily log experiments and parallelize hyperparameter search for neural networks
  ☆735Updated 2 years ago
• szagoruyko / functional-zoo
  PyTorch and Tensorflow functional model definitions
  ☆586Updated 6 years ago
• t-vi / pytorch-tvmisc
  Totally Versatile Miscellanea for Pytorch
  ☆461Updated 2 years ago
• mgrankin / over9000
  Over9000 optimizer
  ☆424Updated last year
• msamogh / nonechucks
  Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!
  ☆375Updated last year
• achaiah / pywick
  High-level batteries-included neural network training library for Pytorch
  ☆397Updated 2 years ago
• moemen95 / Pytorch-Project-Template
  A scalable template for PyTorch projects, with examples in Image Segmentation, Object classification, GANs and Reinforcement Learning.
  ☆865Updated 11 months ago
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆635Updated last year
• GRAAL-Research / poutyne
  A simplified framework and utilities for PyTorch
  ☆570Updated 2 months ago
• szymonmaszke / torchlayers
  Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
  ☆568Updated 2 years ago
• Santosh-Gupta / SpeedTorch
  Library for faster pinned CPU <-> GPU transfer in Pytorch
  ☆682Updated 4 years ago