ayulockin / debugNNwithWandBLinks
Concepts Explored in/with Pytorch
☆20Updated last year
Alternatives and similar repositories for debugNNwithWandB
Users that are interested in debugNNwithWandB are comparing it to the libraries listed below
Sorting:
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Module 0 - Fundamentals☆110Updated last year
- ☆36Updated 3 years ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 9 months ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- NumPy-style histograms in PyTorch☆54Updated 4 months ago
- Convert scikit-learn models to PyTorch modules☆168Updated last year
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆132Updated last year
- A Python library for Continual Inference Networks in PyTorch☆54Updated 10 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Updated last month
- Running Jax in PyTorch Lightning☆119Updated last year
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 3 years ago
- 11-785 Introduction to Deep Learning (IDeeL) website with logistics and select course materials☆81Updated this week
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆37Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated 2 years ago
- MinT: Minimal Transformer Library and Tutorials☆260Updated 3 years ago
- PyTorch Tutorial at the LOD2021 conference☆21Updated 4 years ago
- Official code for the Stochastic Polyak step-size optimizer☆139Updated last year
- Simply Numpy implementation of the FAVOR+ attention mechanism, https://teddykoker.com/2020/11/performers/☆38Updated 5 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Updated last year
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 5 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆75Updated 7 months ago
- A tiny library for stochastic dataset caching in PyTorch.☆45Updated last year
- JAX implementation of Learning to learn by gradient descent by gradient descent☆28Updated 6 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆98Updated 4 years ago
- Unofficial JAX implementations of deep learning research papers☆161Updated 3 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆76Updated 2 years ago
- This is a port of Mistral-7B model in JAX☆33Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆43Updated 4 years ago