emaballarin / ebtorch
🤓🔥 Collection of PyTorch additions, extensions, utilities, uses and abuses
☆9Updated last month
Related projects: ⓘ
- FID computation in Jax/Flax.☆23Updated 2 months ago
- Fast training of unitary deep network layers from low-rank updates☆28Updated last year
- ☆18Updated 5 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆45Updated last month
- Quantification of Uncertainty with Adversarial Models☆27Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆29Updated last year
- Repository for the PopulAtion Parameter Averaging (PAPA) paper☆26Updated 5 months ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 2 years ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆13Updated 3 weeks ago
- ☆21Updated this week
- ☆17Updated 4 months ago
- ☆23Updated this week
- ☆12Updated 9 months ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆19Updated 4 years ago
- ☆13Updated this week
- ☆20Updated this week
- Implementation of a holodeck, written in Pytorch☆17Updated 10 months ago
- Open source code for EigenGame.☆28Updated last year
- Automatically take good care of your preemptible TPUs☆28Updated last year
- Personal solutions to the Triton Puzzles☆11Updated 2 months ago
- Running Jax in PyTorch Lightning☆76Updated last year
- Image augmentation library for Jax☆36Updated 5 months ago
- ☆16Updated this week
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- ☆11Updated this week
- Load any clip model with a standardized interface☆21Updated 4 months ago
- ☆28Updated last week
- ☆24Updated this week
- ☆56Updated 2 years ago