cognizant-ai-labs / autoinit
AutoInit: Analytic Signal-Preserving Weight Initialization for Neural Networks
☆27Updated 2 years ago
Alternatives and similar repositories for autoinit:
Users that are interested in autoinit are comparing it to the libraries listed below
- Implementation of the ALI-G algorithm (PyTorch, Tensorflow)☆22Updated 4 years ago
- Tensorflow 2.x implementation of Gradient Origin Networks☆12Updated 4 years ago
- Implementation of the LOSSGRAD optimization algorithm☆15Updated 6 years ago
- ☆25Updated 4 years ago
- Successfully training approximations to full-rank matrices for efficiency in deep learning.☆17Updated 4 years ago
- A PyTorch Dataset that caches samples in shared memory, accessible globally to all processes☆20Updated 2 years ago
- NeurIPS 2019 Paper Implementation☆12Updated 2 years ago
- Universal Python binding for the LMDB 'Lightning' Database☆13Updated 7 years ago
- Utilities for sequential processing of tar files.☆24Updated 3 years ago
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- Implementation for NATv2.☆23Updated 4 years ago
- Partially Adaptive Momentum Estimation method in the paper "Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep …☆39Updated 2 years ago
- PyTorch ObjectDetection Modules and ONNX ops☆18Updated last year
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks☆18Updated 5 years ago
- Lambda Networks implemented in PyTorch☆13Updated 4 years ago
- Implementation of Kronecker Attention in Pytorch☆18Updated 4 years ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆15Updated 3 years ago
- ☆13Updated 7 years ago
- Convolutions and more as einsum for PyTorch☆16Updated 11 months ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 6 months ago
- ☆41Updated 4 years ago
- Code for "MIM: Mutual Information Machine" paper.☆16Updated 2 years ago
- Repository for the code of the paper "Neural Networks Regularization Through Class-wise Invariant Representation Learning".☆12Updated 7 years ago
- ☆14Updated 5 years ago
- Code for "Training Generative Adversarial Networks with Binary Neurons by End-to-end Backpropagation"☆26Updated 5 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆10Updated 2 years ago
- This tool dumps images in tensorboard☆17Updated 4 years ago
- Exploiting Uncertainty of Loss Landscape for Stochastic Optimization☆15Updated 5 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆17Updated last year