pytorch / extension-ffi

Related projects ⓘ

Alternatives and complementary repositories for extension-ffi

• szagoruyko / pyinn
  CuPy fused PyTorch neural networks ops
  ☆274Updated 6 years ago
• csarofeen / examples
  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  ☆39Updated 6 years ago
• sniklaus / pytorch-extension
  an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors
  ☆117Updated 8 months ago
• davidstutz / tensorflow-cpp-op-example
  Simple example of implementing a new Tensorflow operation and its gradient in C++.
  ☆56Updated 5 years ago
• openai / weightnorm
  Example code for Weight Normalization, from "Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Netw…
  ☆365Updated 5 years ago
• JianGoForIt / YellowFin_Pytorch
  auto-tuning momentum SGD optimizer
  ☆287Updated 5 years ago
• dmlc / mxnet-memonger
  Sublinear memory optimization for deep learning, reduce GPU memory cost to train deeper nets
  ☆308Updated 7 years ago
• noahfl / densenet-sdr
  repo that holds code for improving on dropout using Stochastic Delta Rule
  ☆142Updated 5 years ago
• szagoruyko / functional-zoo
  PyTorch and Tensorflow functional model definitions
  ☆586Updated 6 years ago
• renmengye / revnet-public
  Code for "The Reversible Residual Network: Backpropagation Without Storing Activations"
  ☆351Updated 6 years ago
• uber-research / sbnet
  Sparse Blocks Networks
  ☆436Updated 6 years ago
• eladhoffer / bigBatch
  Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…
  ☆148Updated 7 years ago
• rahulkidambi / AccSGD
  Implements pytorch code for the Accelerated SGD algorithm.
  ☆214Updated 6 years ago
• tensorpack / benchmarks
  Use TensorFlow efficiently
  ☆95Updated 3 years ago
• KarenUllrich / Tutorial-SoftWeightSharingForNNCompression
  A tutorial on 'Soft weight-sharing for Neural Network compression' published at ICLR2017
  ☆146Updated 7 years ago
• yaroslavvb / memory_util
  TensorFlow util for building memory usage timeline from LOG_MEMORY messages
  ☆65Updated 6 years ago
• BayesWatch / sequential-imagenet-dataloader
  A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.
  ☆238Updated 3 years ago
• NVIDIA / pynvrtc
  Python Binding to NVRTC
  ☆79Updated last month
• xgastaldi / shake-shake
  2.86% and 15.85% on CIFAR-10 and CIFAR-100
  ☆295Updated 6 years ago
• philkr / magic_init
  ☆138Updated 8 years ago
• bes-dev / mpl.pytorch
  Pytorch implementation of MaxPoolingLoss.
  ☆175Updated 6 years ago
• Lyken17 / SparseNet
  [ECCV 2018] Sparsely Aggreagated Convolutional Networks https://arxiv.org/abs/1801.05895
  ☆125Updated 6 years ago
• dmlc / tensorboard
  Standalone TensorBoard for visualizing in deep learning
  ☆370Updated 4 years ago
• chrischoy / MakePytorchPlusPlus
  How and why you want to make your pytorch CUDA/CPP extension with a Makefile
  ☆171Updated 5 years ago
• hpi-xnor / BMXNet
  (New version is out: https://github.com/hpi-xnor/BMXNet-v2) BMXNet: An Open-Source Binary Neural Network Implementation Based on MXNet
  ☆351Updated 4 years ago
• fmassa / optimize-net
  OptNet - Reducing memory usage in torch neural nets
  ☆282Updated 7 years ago
• chengtaipu / lowrankcnn
  Low-rank convolutional neural networks
  ☆96Updated 8 years ago
• apaszke / torch-opCounter
  ☆28Updated 6 years ago
• szagoruyko / diracnets
  Training Very Deep Neural Networks Without Skip-Connections
  ☆585Updated 6 years ago
• TeamHG-Memex / tensorboard_logger
  Log TensorBoard events without touching TensorFlow
  ☆630Updated last year