Alternatives and similar repositories for Sharpness-Aware-Minimization-TensorFlow:

Users that are interested in Sharpness-Aware-Minimization-TensorFlow are comparing it to the libraries listed below

• sayakpaul / MLP-Mixer-CIFAR10
  Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.
  ☆54Updated 2 years ago
• simon-larsson / keras-swa
  Simple stochastic weight averaging callback for Keras
  ☆63Updated 3 years ago
• ayulockin / LossLandscape
  Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …
  ☆62Updated 4 years ago
• aiwabdn / pygln
  Python implementation of GLN in different frameworks
  ☆97Updated 4 years ago
• sayakpaul / Adaptive-Gradient-Clipping
  Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.
  ☆83Updated 3 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• yk / PyTorch_CIFAR10
  Pretrained TorchVision models on CIFAR10 dataset (with weights)
  ☆24Updated 4 years ago
• dhuynh95 / fastai_bayesian
  Quick modules to turn regular Neural Networks to Bayesian Neural Networks with Dropout.
  ☆35Updated 3 years ago
• ariG23498 / G-SimCLR
  This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip…
  ☆76Updated 3 years ago
• ayulockin / DataAugmentationTF
  Implementation of modern data augmentation techniques in TensorFlow 2.x to be used in your training pipeline.
  ☆34Updated 4 years ago
• jxbz / madam
  👩 Pytorch and Jax code for the Madam optimiser.
  ☆51Updated 4 years ago
• AakashKumarNain / AugMix_TF2
  TF2.0 port for Augmix paper
  ☆79Updated 5 years ago
• rwightman / efficientnet-jax
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆127Updated last year
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆123Updated 7 months ago
• kartik4949 / TensorPipe
  High Performance Tensorflow Data Pipeline with State of Art Augmentations and low level optimizations.
  ☆86Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆106Updated 2 years ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 2 years ago
• g-benton / learning-invariances
  Codebase for Learning Invariances in Neural Networks
  ☆93Updated 2 years ago
• beresandras / contrastive-classification-keras
  Implementation of self-supervised image-level contrastive pretraining methods using Keras.
  ☆69Updated 3 years ago
• sayakpaul / PAWS-TF
  Minimal implementation of PAWS (https://arxiv.org/abs/2104.13963) in TensorFlow.
  ☆45Updated 3 years ago
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆213Updated last year
• IssamLaradji / sps
  Official code for the Stochastic Polyak step-size optimizer
  ☆138Updated 7 months ago
• ganguli-lab / degrees-of-freedom
  ☆36Updated 3 years ago
• nestordemeure / ManifoldMixup
  Manifold-Mixup implementation for fastai V1
  ☆19Updated 4 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 4 years ago
• awaelchli / pytorch-lightning-snippets
  A collection of code snippets for my PyTorch Lightning projects
  ☆107Updated 4 years ago
• uber-research / GTN
  ☆77Updated 4 years ago
• kyunghyuncho / jax-practice
  ☆12Updated 4 years ago
• srush / parallax
  ☆153Updated 4 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 3 years ago