cedricrommel / augnet
Code of "Deep invariant networks with differentiable augmentation layers"
☆18Updated 2 years ago
Alternatives and similar repositories for augnet:
Users that are interested in augnet are comparing it to the libraries listed below
- A project to improve out-of-distribution detection (open set recognition) and uncertainty estimation by changing a few lines of code in y…☆45Updated 2 years ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆17Updated 3 months ago
- ☆51Updated 8 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- ☆33Updated last year
- ☆47Updated 2 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆54Updated 2 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆27Updated 3 years ago
- FID computation in Jax/Flax.☆26Updated 6 months ago
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 10 months ago
- Implementation of LogAvgExp for Pytorch☆33Updated 2 years ago
- Latest Weight Averaging (NeurIPS HITY 2022)☆28Updated last year
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- Unsupervised diverse image generation via GANs: Partition Guided Mixture of Generative Adversarial Networks☆12Updated 3 years ago
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 5 months ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Code for "Invariance Learning in Deep Neural Networks with Differentiable Laplace Approximations"☆23Updated 2 years ago
- Convolutions and more as einsum for PyTorch☆14Updated 8 months ago
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆60Updated 4 months ago
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- ☆31Updated 3 years ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- Official repository for MaGNET, ICLR 2022☆24Updated 2 years ago
- Personal implementation of ASIF by Antonio Norelli☆25Updated 8 months ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Updated last year
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 3 weeks ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆57Updated last year