ropas / pytea
PyTea: PyTorch Tensor shape error analyzer
☆315Updated 2 years ago
Alternatives and similar repositories for pytea:
Users that are interested in pytea are comparing it to the libraries listed below
- Deep Learning project template for PyTorch (multi-gpu training is supported)☆135Updated last year
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆773Updated 7 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆298Updated 3 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆292Updated 4 years ago
- The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…☆802Updated 2 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆596Updated 9 months ago
- Profiling and inspecting memory in pytorch☆1,042Updated 6 months ago
- Library for 8-bit optimizers and quantization routines.☆717Updated 2 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,029Updated 10 months ago
- Demystify RAM Usage in Multi-Process Data Loaders☆187Updated last year
- Visualize PyTorch tensors with a single line of code.☆657Updated last month
- Pytorch Lightning code guideline for conferences☆1,253Updated last year
- functorch is JAX-like composable function transforms for PyTorch.☆1,410Updated this week
- optimizer & lr scheduler & loss function collections in PyTorch☆269Updated this week
- Flexible components pairing 🤗 Transformers with Pytorch Lightning☆612Updated 2 years ago
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆226Updated 2 years ago
- Official Code for "Non-deep Networks"☆585Updated 2 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆215Updated last week
- Learning Rate Warmup in PyTorch☆403Updated 2 weeks ago
- An All-MLP solution for Vision, from Google AI☆1,013Updated 5 months ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆873Updated 5 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆419Updated 9 months ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,165Updated this week
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆226Updated last month
- Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.☆233Updated 2 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,064Updated 2 years ago
- Ranger deep learning optimizer rewrite to use newest components☆327Updated last year
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆210Updated 2 years ago
- Code release for "Dropout Reduces Underfitting"☆311Updated last year
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆369Updated last year