NVIDIA / framework-reproducibility
Providing reproducibility in deep learning frameworks
☆425Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for framework-reproducibility
- Keras/TF implementation of AdamW, SGDW, NadamW, Warm Restarts, and Learning Rate multipliers☆167Updated 2 years ago
- A GPU performance profiling tool for PyTorch models☆495Updated 3 years ago
- Fast Block Sparse Matrices for Pytorch☆545Updated 3 years ago
- RAdam implemented in Keras & TensorFlow☆325Updated 2 years ago
- A profiling and performance analysis tool for TensorFlow☆360Updated this week
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆317Updated last year
- ☆302Updated 3 years ago
- Experimental ground for optimizing memory of pytorch models☆360Updated 6 years ago
- Implementation for the Lookahead Optimizer.☆239Updated 2 years ago
- Implementations of ideas from recent papers☆391Updated 3 years ago
- Large Model Support in Tensorflow☆202Updated 4 years ago
- Guide for building custom op for TensorFlow☆378Updated last year
- Long Range Arena for Benchmarking Efficient Transformers☆729Updated 11 months ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆695Updated last year
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Updated 4 years ago
- Implementation of https://arxiv.org/abs/1904.00962☆369Updated 3 years ago
- Fast Differentiable Sorting and Ranking☆574Updated 9 months ago
- This is a Tensor Train based compression library to compress sparse embedding tables used in large-scale machine learning models such as …☆193Updated 2 years ago
- Web-based dashboard for Sacred☆547Updated last year
- Profiling and inspecting memory in pytorch☆1,020Updated 3 months ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- ThunderGBM: Fast GBDTs and Random Forests on GPUs☆694Updated 9 months ago
- Deep Learning Experiment Management☆639Updated last year
- A High Level API for Deep Learning in JAX☆470Updated last year
- PyTorch layer-by-layer model profiler☆608Updated 3 years ago
- Dataset, streaming, and file system extensions maintained by TensorFlow SIG-IO☆706Updated this week
- Robust Bi-Tempered Logistic Loss Based on Bregman Divergences. https://arxiv.org/pdf/1906.03361.pdf☆147Updated 2 years ago
- Decoupled Weight Decay Regularization (ICLR 2019)☆265Updated 5 years ago
- Python library to easily log experiments and parallelize hyperparameter search for neural networks☆736Updated 2 years ago
- Over9000 optimizer☆425Updated last year