AminRezaei0x443 / PyTorch-LIT
Lite Inference Toolkit (LIT) for PyTorch
☆161Updated 3 years ago
Alternatives and similar repositories for PyTorch-LIT:
Users that are interested in PyTorch-LIT are comparing it to the libraries listed below
- Python Research Framework☆106Updated 2 years ago
- My implementation of DeepMind's Perceiver☆63Updated 3 years ago
- XtremeDistil framework for distilling/compressing massive multilingual neural network models to tiny and efficient models for AI at scale☆154Updated last year
- A library to inspect and extract intermediate layers of PyTorch models.☆472Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Code for the Shortformer model, from the ACL 2021 paper by Ofir Press, Noah A. Smith and Mike Lewis.☆146Updated 3 years ago
- Implementation of Feedback Transformer in Pytorch☆105Updated 4 years ago
- The "tl;dr" on a few notable transformer papers (pre-2022).☆190Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆183Updated 2 years ago
- Functional deep learning☆108Updated 2 years ago
- MinT: Minimal Transformer Library and Tutorials☆253Updated 2 years ago
- Accelerated NLP pipelines for fast inference on CPU. Built with Transformers and ONNX runtime.☆126Updated 4 years ago
- A simple library that implements CLIP guided loss in PyTorch.☆77Updated 3 years ago
- Management Dashboard for Torchserve☆120Updated 2 years ago
- Open Source Photos Platform Powered by PyTorch☆135Updated 2 years ago
- Check if you have training samples in your test set☆64Updated 2 years ago
- RUDOLPH: One Hyper-Tasking Transformer can be creative as DALL-E and GPT-3 and smart as CLIP☆254Updated 2 years ago
- An alternative to convolution in neural networks☆254Updated last year
- Library for 8-bit optimizers and quantization routines.☆716Updated 2 years ago
- Accelerated NLP pipelines for fast inference on CPU and GPU. Built with Transformers, Optimum and ONNX Runtime.☆125Updated 3 years ago
- Babysit your preemptible TPUs☆85Updated 2 years ago
- A fastai/PyTorch package for unpaired image-to-image translation.☆135Updated last year
- FasterAI: Prune and Distill your models with FastAI and PyTorch☆247Updated last week
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 2 years ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 4 months ago
- Adaptation of discriminative learning rates from the Fastai library for standard PyTorch.☆18Updated 5 years ago
- DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight …☆235Updated last year
- State of the art faster Transformer with Tensorflow 2.0 ( NLP, Computer Vision, Audio ).☆85Updated 2 years ago
- Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)☆116Updated 3 years ago
- HetSeq: Distributed GPU Training on Heterogeneous Infrastructure☆106Updated last year