Alternatives and similar repositories for pytea

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

• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆194Updated 2 years ago
• ryul99 / pytorch-project-template
  Deep Learning project template for PyTorch (multi-gpu training is supported)
  ☆138Updated last year
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆299Updated 2 weeks ago
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆291Updated 4 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…
  ☆808Updated 3 years ago
• rish-16 / aft-pytorch
  Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.
  ☆237Updated 3 years ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,074Updated 2 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆346Updated 5 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆792Updated 2 weeks ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆218Updated 3 months ago
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆715Updated 2 years ago
• Lightning-Universe / lightning-transformers
  Flexible components pairing 🤗 Transformers with Pytorch Lightning
  ☆609Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,025Updated 9 months ago
• imankgoyal / NonDeepNetworks
  Official Code for "Non-deep Networks"
  ☆585Updated 2 years ago
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆225Updated 3 years ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,432Updated this week
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆617Updated last year
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆480Updated 2 years ago
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,061Updated 10 months ago
• pytorch / torchdynamo
  A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
  ☆1,053Updated last year
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆331Updated last year
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆313Updated 2 years ago
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆639Updated 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
• huggingface / nn_pruning
  Prune a model while finetuning or training.
  ☆403Updated 3 years ago
• EfficientDL / book
  PDFs and Codelabs for the Efficient Deep Learning book.
  ☆193Updated 2 years 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 last year
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆410Updated last week
• SforAiDl / KD_Lib
  A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…
  ☆634Updated 2 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆817Updated 2 years ago