facebookresearch / madgrad

Related projects ⓘ

Alternatives and complementary repositories for madgrad

• szymonmaszke / torchlayers
  Shape and dimension inference (Keras-like) for PyTorch layers and neural networks
  ☆568Updated 2 years ago
• Lightning-Universe / lightning-flash
  Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains
  ☆1,743Updated last year
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,131Updated this week
• Lightning-Universe / lightning-bolts
  Toolbox of models, callbacks, and datasets for AI/ML researchers.
  ☆1,691Updated this week
• juntang-zhuang / Adabelief-Optimizer
  Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"
  ☆1,050Updated 3 months ago
• archinetai / surgeon-pytorch
  A library to inspect and extract intermediate layers of PyTorch models.
  ☆470Updated 2 years ago
• f-dangel / cockpit
  Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
  ☆474Updated 2 years ago
• grok-ai / nn-template
  Generic template to bootstrap your PyTorch project.
  ☆635Updated last year
• google / objax
  ☆771Updated 9 months ago
• patrick-kidger / torchtyping
  Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
  ☆1,402Updated 3 months ago
• google / ml_collections
  ML Collections is a library of Python Collections designed for ML use cases.
  ☆893Updated 3 months ago
• jonasrauber / eagerpy
  PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code
  ☆696Updated last year
• PhoenixDL / rising
  Provides everything needed for high performance data loading and augmentation in pytorch.
  ☆315Updated 10 months ago
• KeremTurgutlu / self_supervised
  Implementation of popular SOTA self-supervised learning algorithms as Fastai Callbacks.
  ☆318Updated last year
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆714Updated 2 years ago
• brohrer / sharpened-cosine-similarity
  An alternative to convolution in neural networks
  ☆250Updated 7 months ago
• OATML / non-parametric-transformers
  Code for "Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning"
  ☆404Updated 7 months ago
• poets-ai / elegy
  A High Level API for Deep Learning in JAX
  ☆469Updated last year
• replicate / keepsake
  Version control for machine learning
  ☆1,650Updated 2 months ago
• pytorch / nestedtensor
  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  ☆253Updated last year
• szymonmaszke / torchdatasets
  PyTorch dataset extended with map, cache etc. (tensorflow.data like)
  ☆328Updated 2 years ago
• cresset-template / cresset
  Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.
  ☆715Updated 2 weeks ago
• Lightning-Universe / lightning-transformers
  Flexible components pairing 🤗 Transformers with Pytorch Lightning
  ☆611Updated last year
• huggingface / pytorch_block_sparse
  Fast Block Sparse Matrices for Pytorch
  ☆545Updated 3 years ago
• lkhphuc / lightning-hydra-template
  Deep Learning project template best practices with Pytorch Lightning, Hydra, Tensorboard.
  ☆155Updated 3 years ago
• libffcv / ffcv
  FFCV: Fast Forward Computer Vision (and other ML workloads!)
  ☆2,857Updated 4 months ago
• NVIDIA-Merlin / dataloader
  The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX
  ☆407Updated 6 months ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆775Updated 10 months ago
• airctic / icevision
  An Agnostic Computer Vision Framework - Pluggable to any Training Library: Fastai, Pytorch-Lightning with more to come
  ☆849Updated this week
• BlackHC / toma
  Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
  ☆426Updated 2 months ago