Alternatives and similar repositories for pytea

Users that are interested in pytea are comparing it to the libraries listed below

• ryul99 / pytorch-project-template
  Deep Learning project template for PyTorch (multi-gpu training is supported)
  ☆138Updated 2 years ago
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆300Updated 2 months ago
• lucidrains / memory-efficient-attention-pytorch
  Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"
  ☆379Updated 2 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆796Updated 2 months ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆197Updated 2 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆221Updated 5 months ago
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆225Updated 3 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆297Updated last year
• pytorch / torcheval
  A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…
  ☆236Updated last week
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆334Updated last year
• pytorch / ort
  Accelerate PyTorch models with ONNX Runtime
  ☆363Updated 6 months ago
• lucidrains / local-attention
  An implementation of local windowed attention for language modeling
  ☆474Updated last month
• AminRezaei0x443 / memory-efficient-attention
  Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
  ☆184Updated 2 years ago
• fgaim / gpuview
  A lightweight web dashboard for monitoring GPU usage
  ☆201Updated last year
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆251Updated 2 years ago
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆314Updated 2 years ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆290Updated 4 years ago
• lucidrains / FLASH-pytorch
  Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"
  ☆369Updated last year
• EfficientDL / book
  PDFs and Codelabs for the Efficient Deep Learning book.
  ☆195Updated 2 years ago
• vra / flopth
  A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.
  ☆130Updated 9 months ago
• huggingface / nn_pruning
  Prune a model while finetuning or training.
  ☆403Updated 3 years ago
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆777Updated 3 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆795Updated last year
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆215Updated 2 years ago
• pytorch / hydra-torch
  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  ☆221Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,037Updated last month
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆273Updated 3 years ago
• DeMoriarty / TorchPQ
  Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda
  ☆226Updated last year
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆504Updated 4 years ago