Alternatives and similar repositories for GradNorm

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

• AvivNavon / nash-mtl
  Official implementation of "Multi-Task Learning as a Bargaining Game" [ICML 2022]
  ☆236Updated 7 months ago
• Cranial-XIX / FAMO
  Official PyTorch Implementation for Fast Adaptive Multitask Optimization (FAMO)
  ☆116Updated last year
• WeiChengTseng / Pytorch-PCGrad
  Pytorch reimplementation for "Gradient Surgery for Multi-Task Learning"
  ☆394Updated 4 years ago
• deebuls / deep_evidential_regression_loss_pytorch
  Implementation of Deep evidential regression paper
  ☆58Updated 5 years ago
• hurrialice / uncertainties
  AM207 project: dissect aleatoric and epistemic uncertainty
  ☆90Updated 6 years ago
• bentherien / parametricScatteringNetworks
  [Oral; CVPR'22] Parametric Scattering Networks
  ☆28Updated 2 months ago
• brianlan / pytorch-grad-norm
  Pytorch implementation of the GradNorm. GradNorm addresses the problem of balancing multiple losses for multi-task learning by learning a…
  ☆271Updated 3 years ago
• lorenmt / auto-lambda
  The Implementation of "Auto-Lambda: Disentangling Dynamic Task Relationships" [TMLR 2022].
  ☆140Updated 3 years ago
• unnir / cVAE
  Simple and clean implementation of Conditional Variational AutoEncoder (cVAE) using PyTorch
  ☆127Updated 2 years ago
• hosseinshn / GradNorm
  This in my Demo of Chen et al. "GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks" ICML 2018
  ☆181Updated 4 years ago
• hmi88 / what
  Pytorch implementation of "What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?"
  ☆170Updated 6 years ago
• teddykoker / evidential-learning-pytorch
  Evidential Deep Learning in PyTorch
  ☆68Updated 3 years ago
• Lingkai-Kong / SDE-Net
  Code for paper: SDE-Net: Equipping Deep Neural network with Uncertainty Estimates
  ☆117Updated 5 years ago
• EnnengYang / AdaTask
  AdaTask: A Task-Aware Adaptive Learning Rate Approach to Multi-Task Learning. AAAI, 2023.
  ☆29Updated 2 years ago
• ldeecke / gmm-torch
  Gaussian mixture models in PyTorch.
  ☆584Updated 2 years ago
• cpark321 / uncertainty-deep-learning
  ☆239Updated 5 years ago
• Cranial-XIX / CAGrad
  Official PyTorch Implementation for Conflict-Averse Gradient Descent (CAGrad)
  ☆144Updated 2 years ago
• Hui-Li / multi-task-learning-example-PyTorch
  ☆148Updated 4 years ago
• JakobCode / UncertaintyInNeuralNetworks_Resources
  ☆110Updated 4 years ago
• xxgege / StableNet
  Official repository for CVPR21 paper "Deep Stable Learning for Out-Of-Distribution Generalization".
  ☆200Updated 3 years ago
• xiaofangxd / Multi-objective-optimization-and-Multi-task-Learning
  ☆72Updated last year
• cure-lab / deep-active-learning
  An implementation of the state-of-the-art Deep Active Learning algorithms
  ☆107Updated 2 years ago
• thuml / Flowformer
  About Code release for "Flowformer: Linearizing Transformers with Conservation Flows" (ICML 2022), https://arxiv.org/pdf/2202.06258.pdf
  ☆331Updated last year
• huyng / incertae
  Code for MC Dropout and Model Ensembling Uncertainty Estimate experiments
  ☆71Updated 5 years ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆216Updated 4 years ago
• ucals / cvae
  ☆73Updated 2 years ago
• mit-gfx / ContinuousParetoMTL
  [ICML 2020] Efficient Continuous Pareto Exploration in Multi-Task Learning
  ☆149Updated 4 years ago
• ElektrischesSchaf / LayerNorm_GRU
  Implement layer normalization GRU in pytorch
  ☆35Updated 2 years ago
• Flawless1202 / Non-AR-Spatial-Temporal-Transformer
  Implementation of the paper NAST: Non-Autoregressive Spatial-Temporal Transformer for Time Series Forecasting.
  ☆77Updated 4 years ago
• aamini / evidential-deep-learning
  Learn fast, scalable, and calibrated measures of uncertainty using neural networks!
  ☆508Updated 4 years ago