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 last year
• 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…
  ☆806Updated 3 years ago
• zhijian-liu / torchprofile
  A general and accurate MACs / FLOPs profiler for PyTorch models
  ☆604Updated last year
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆716Updated 2 years ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆329Updated last year
• kaiyuyue / torchshard
  Slicing a PyTorch Tensor Into Parallel Shards
  ☆298Updated 3 years ago
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆790Updated 10 months ago
• NVIDIA / runx
  Deep Learning Experiment Management
  ☆639Updated 2 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆218Updated last month
• ppwwyyxx / RAM-multiprocess-dataloader
  Demystify RAM Usage in Multi-Process Data Loaders
  ☆193Updated 2 years ago
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,424Updated this week
• huggingface / nn_pruning
  Prune a model while finetuning or training.
  ☆402Updated 2 years ago
• patrick-kidger / torchtyping
  Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
  ☆1,432Updated last week
• ucbrise / actnn
  ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
  ☆200Updated 2 years ago
• imankgoyal / NonDeepNetworks
  Official Code for "Non-deep Networks"
  ☆585Updated 2 years ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆251Updated 2 years ago
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆225Updated 3 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,021Updated 7 months ago
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆420Updated 11 months ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,071Updated 2 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆815Updated 2 years ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆345Updated 5 years ago
• cmsflash / efficient-attention
  An implementation of the efficient attention module.
  ☆310Updated 4 years ago
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆476Updated 2 years ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆320Updated last year
• SHI-Labs / Neighborhood-Attention-Transformer
  Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
  ☆1,115Updated 11 months ago
• facebookresearch / dropout
  Code release for "Dropout Reduces Underfitting"
  ☆313Updated 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
• SforAiDl / KD_Lib
  A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…
  ☆625Updated 2 years ago