ayulockin / debugNNwithWandBLinks
Concepts Explored in/with Pytorch
☆19Updated last year
Alternatives and similar repositories for debugNNwithWandB
Users that are interested in debugNNwithWandB are comparing it to the libraries listed below
Sorting:
- 11-785 Introduction to Deep Learning (IDeeL) website with logistics and select course materials☆72Updated last week
- JAX implementation of Learning to learn by gradient descent by gradient descent☆28Updated 5 months ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- ☆36Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 5 years ago
- A simple Transformer where the softmax has been replaced with normalization☆20Updated 5 years ago
- PyTorch Tutorial at the LOD2021 conference☆21Updated 4 years ago
- ☆75Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 4 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
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 5 years ago
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆43Updated 4 years ago
- reproduces experiments from "Grounding inductive biases in natural images: invariance stems from variations in data"☆17Updated last year
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 3 years ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated last year
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆67Updated 3 weeks ago
- ☆39Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- A JAX nn library☆22Updated 4 months ago
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆46Updated 4 years ago
- Awesome list for Neural Network Optimization methods.☆81Updated 10 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated last year
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆39Updated 4 years ago
- Meta Optimal Transport☆105Updated 2 years ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆97Updated 4 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 8 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆124Updated 4 months ago