samrere / pytorttoLinks
deep learning from scratch. uses numpy/cupy, trains in GPU, follows pytorch API
☆21Updated 2 years ago
Alternatives and similar repositories for pytortto
Users that are interested in pytortto are comparing it to the libraries listed below
Sorting:
- A Numpy implementation of a Convolutional Neural Network: slow & fast (im2col/col2im).☆58Updated 2 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- How to train a CNN to 99% accuracy on MNIST in less than a second on a laptop☆70Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- Deep Learning book the covers the principles of deep learning, motivation, explanations, state of the art papers for the various tasks an…☆107Updated 3 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated last year
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆245Updated 2 weeks ago
- This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.☆192Updated 4 years ago
- Neural Networks library in pure numpy☆70Updated last year
- Spio (SPEE-oh) - Experimental CUDA kernel framework unifying typed dimensions, NVRTC JIT specialization, and ML‑guided tuning.☆46Updated this week
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆88Updated 4 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- Easily benchmark PyTorch model FLOPs, latency, throughput, allocated gpu memory and energy consumption☆109Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 3 years ago
- A crash course on PyTorch hooks☆40Updated 5 years ago
- Learning with Signatures☆57Updated 3 years ago
- Pytorch cyclic cosine decay learning rate scheduler☆49Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 4 years ago
- Customized matrix multiplication kernels☆57Updated 3 years ago
- Simple stochastic weight averaging callback for Keras☆63Updated 4 years ago
- Code for a workshop hosted at the MLOps World Summit '22☆18Updated 3 years ago
- PDFs and Codelabs for the Efficient Deep Learning book.☆203Updated 2 years ago
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 4 years ago
- A Hackable Quantization Library for PyTorch☆22Updated 4 years ago
- ☆51Updated last year
- PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)☆31Updated 4 years ago
- ONNX Command-Line Toolbox☆35Updated last year
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Yet another mini autodiff system for educational purposes☆31Updated last month