ezyang / pytorch-unattached
Tensors and Dynamic neural networks in Python with strong GPU acceleration
☆21Updated last year
Alternatives and similar repositories for pytorch-unattached:
Users that are interested in pytorch-unattached are comparing it to the libraries listed below
- ☆42Updated 5 years ago
- ONNX Integration Builds☆20Updated 6 years ago
- ☆51Updated 4 years ago
- Stride visualizations☆36Updated 6 years ago
- Berg - Run GPU-backed experiments on gcloud☆26Updated 6 years ago
- Implementation of Neural Arithmetic Logic Units (https://arxiv.org/pdf/1808.00508.pdf)☆31Updated 6 years ago
- A Python library for the Sequential Halving and Classification algorithm☆21Updated 2 years ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆70Updated 2 years ago
- ☆19Updated last year
- CURRENNT -- CUDA-enabled machine learning library for recurrent neural network☆16Updated 4 years ago
- Generalized Compressed Network Search with PyTorch☆26Updated 7 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- Notes from Simons Institute program "Foundations of Machine Learning"☆13Updated 7 years ago
- Logging utility for ML experiments☆16Updated 2 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- ☆12Updated 3 years ago
- ☆14Updated 7 years ago
- Implements an infinite sum of poisson-weighted convolutions☆26Updated 6 years ago
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- Implementation of a Tensorflow XLA rematerialization pass☆15Updated 5 years ago
- nGraph™ Backend for ONNX☆42Updated 2 years ago
- ☆12Updated 7 years ago
- 🤖 Implementation of Self Normalizing Networks (SNN) in PyTorch.☆12Updated 7 years ago
- Distributed Learning by Pair-Wise Averaging☆53Updated 7 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- .NET bindings for the Pytorch engine☆17Updated 5 years ago
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- Discrete Object Generation with Reversible Inductive Construction (NeurIPS 2019)☆30Updated 3 years ago