emaballarin / ebtorchLinks
π€π₯ Collection of PyTorch additions, extensions, utilities, uses and abuses
β9Updated this week
Alternatives and similar repositories for ebtorch
Users that are interested in ebtorch are comparing it to the libraries listed below
Sorting:
- Repository for the PopulAtion Parameter Averaging (PAPA) paperβ27Updated last year
- Load any clip model with a standardized interfaceβ21Updated last year
- Quantification of Uncertainty with Adversarial Modelsβ30Updated 2 years ago
- FID computation in Jax/Flax.β28Updated last year
- Implementation of a holodeck, written in Pytorchβ18Updated last year
- Official code for the paper: "Metadata Archaeology"β19Updated 2 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwoβ¦β71Updated last month
- CUDA implementation of autoregressive linear attention, with all the latest research findingsβ44Updated 2 years ago
- Image augmentation library for Jaxβ39Updated last year
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 4 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"β13Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'β38Updated 3 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wiβ¦β51Updated 3 years ago
- 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
- Implementation of Metaformer, but in an autoregressive mannerβ27Updated 3 years ago
- Utilities for PyTorch distributedβ24Updated 5 months ago
- β31Updated last month
- β74Updated 2 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimatorβ31Updated 2 years ago
- β60Updated 3 years ago
- Implementation of OpenAI paper with Simple Noise Scale on Fastai V2β19Updated 4 years ago
- Little article showing how to load pytorch's models with linear memory consumptionβ34Updated 2 years ago
- Layerwise Batch Entropy Regularizationβ23Updated 3 years ago
- β8Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorchβ88Updated 2 years ago
- A GPT, made only of MLPs, in Jaxβ58Updated 4 years ago
- Python library for argument and configuration managementβ54Updated 2 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)β53Updated 4 months ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"β29Updated 5 years ago
- Implementation of LogAvgExp for Pytorchβ36Updated 4 months ago