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 6 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"
  ☆388Updated 2 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆804Updated 6 months ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆223Updated 8 months ago
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆317Updated 2 years ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆205Updated 2 years ago
• AminRezaei0x443 / memory-efficient-attention
  Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch
  ☆184Updated 2 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆302Updated last year
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,068Updated last year
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆228Updated 3 years ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆293Updated 5 years ago
• huggingface / nn_pruning
  Prune a model while finetuning or training.
  ☆404Updated 3 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.
  ☆131Updated last year
• rish-16 / aft-pytorch
  Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.
  ☆243Updated 3 years ago
• parrt / tensor-sensor
  The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflo…
  ☆812Updated 3 years ago
• lucidrains / flash-cosine-sim-attention
  Implementation of fused cosine similarity attention in the same style as Flash Attention
  ☆219Updated 2 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆277Updated 3 years ago
• meta-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…
  ☆244Updated 2 months ago
• EfficientDL / book
  PDFs and Codelabs for the Efficient Deep Learning book.
  ☆203Updated 2 years ago
• fgaim / gpuview
  A lightweight web dashboard for monitoring GPU usage
  ☆205Updated last month
• bobby-he / simplified_transformers
  ☆293Updated last year
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆250Updated 3 years ago
• ray-project / ray_lightning
  Pytorch Lightning Distributed Accelerators using Ray
  ☆215Updated 2 years ago
• Alibaba-MIIL / Solving_ImageNet
  Official PyTorch implementation of the paper: "Solving ImageNet: a Unified Scheme for Training any Backbone to Top Results" (2022)
  ☆193Updated 2 years ago
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆438Updated last year
• facebookresearch / ppuda
  Code for Parameter Prediction for Unseen Deep Architectures (NeurIPS 2021)
  ☆491Updated 2 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆631Updated 4 months ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆338Updated last year
• pytorch / ort
  Accelerate PyTorch models with ONNX Runtime
  ☆368Updated 9 months ago