moellenh's Stars
tqdm/tqdm
:zap: A Fast, Extensible Progress Bar for Python and CLI
garrettj403/SciencePlots
Matplotlib styles for scientific plotting
google-deepmind/dm-haiku
JAX-based neural network library
davda54/sam
SAM: Sharpness-Aware Minimization (PyTorch)
lezcano/geotorch
Constrained optimization toolkit for PyTorch
google-research/sam
ssydasheng/Bayesian_neural_network_papers
Papers for Bayesian-NN
google-deepmind/kfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
SamsungLabs/ASAM
Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.
ermongroup/sliced_score_matching
Code for reproducing results in the sliced score matching paper (UAI 2019)
ssydasheng/FBNN
Code for "Functional variational Bayesian neural networks" (https://arxiv.org/abs/1903.05779)
team-approx-bayes/ivon
IVON optimizer for neural networks based on variational learning.
GongXinyuu/sngan.pytorch
An unofficial Pytorch implementation of SNGAN, achieving IS of 8.21 and FID of 14.21 on CIFAR-10.
team-approx-bayes/fromp
Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"
ppwwyyxx/FRN-on-common-ImageNet-baseline
Filter Response Normalization tested on better ImageNet baselines.
hiyouga/AMP-Regularizer
Code for our paper "Regularizing Neural Networks via Adversarial Model Perturbation", CVPR2021
team-approx-bayes/bayesian-sam
Code for "SAM as an Optimal Relaxation of Bayes", ICLR 2023.
didriknielsen/ds3_abi_exercises
Exercises for the Tutorial on Approximate Bayesian Inference at the Data Science Summer School 2018
yzhwang/jax-multi-gpu-resnet50-example
An example showing how to use jax to train resnet50 on multi-node multi-GPU
emtiyaz/cvi
Code for AISTATS 2017 paper on "Conjugate-Computation Variational Inference"
team-approx-bayes/kpriors
Code for Knowledge-Adaptation Priors based on the NeurIPS 2021 paper by Khan and Swaroop.
devansh20la/LPF-SGD
EmanuelLaude/compot
Composite optimization toolbox: python library for fast composite optimization
team-approx-bayes/memory-perturbation
Code for "The Memory Perturbation Equation: Understanding Model’s Sensitivity to Data", NeurIPS 2023.
team-approx-bayes/liegroups
Code for "The Lie-Group Bayesian Learning Rule", AISTATS 2023.
team-approx-bayes/ivon-experiments
rioyokotalab/hpsc-2022
ysngshn/ivon-optax
An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)
kigo24/AdamB
ysngshn/ojax
OJAX is a small extension of JAX to facilitate modular coding.