ropas / pytea

Related projects: ⓘ

• ryul99 / pytorch-project-template
  Deep Learning project template for PyTorch (multi-gpu training is supported)
  ☆134Updated last year
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆295Updated 3 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"
  ☆355Updated last year
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆713Updated 2 years ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,057Updated last year
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆746Updated 2 months ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆557Updated 4 months ago
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆221Updated 2 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆244Updated 8 months ago
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆171Updated last year
• Stonesjtu / pytorch_memlab
  Profiling and inspecting memory in pytorch
  ☆1,013Updated last month
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆987Updated this week
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆332Updated 4 years ago
• kozistr / pytorch_optimizer
  optimizer & lr scheduler & loss function collections in PyTorch
  ☆221Updated last week
• PhilJd / contiguous_pytorch_params
  Accelerate training by storing parameters in one contiguous chunk of memory.
  ☆291Updated 3 years ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆322Updated 7 months ago
• rish-16 / aft-pytorch
  Unofficial PyTorch implementation of Attention Free Transformer (AFT) layers by Apple Inc.
  ☆226Updated 2 years ago
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆480Updated 3 years ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆387Updated 4 months ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆207Updated this week
• SforAiDl / KD_Lib
  A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…
  ☆594Updated last year
• thuml / depyf
  depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.
  ☆424Updated this week
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆311Updated last year
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• sacmehta / delight
  DeLighT: Very Deep and Light-Weight Transformers
  ☆465Updated 3 years ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆433Updated last year
• sail-sg / poolformer
  PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)
  ☆1,281Updated 3 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆672Updated 4 months ago
• google-research / vmoe
  ☆557Updated this week
• lessw2020 / Best-Deep-Learning-Optimizers
  Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable
  ☆212Updated 3 years ago
• 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…
  ☆213Updated this week