mlbench / mlbench-coreLinks
MLBench Framework Core Python Library
☆18Updated 2 years ago
Alternatives and similar repositories for mlbench-core
Users that are interested in mlbench-core are comparing it to the libraries listed below
Sorting:
- PyTorch-MPI-DDP-example☆18Updated 7 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 5 months ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆109Updated 2 years ago
- Template repository for a Python 3-based data science project that uses Horovod.☆43Updated 4 years ago
- PyTorch Sphinx Theme☆104Updated 2 weeks ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- PyTorch interface for the IPU☆181Updated 2 years ago
- Example python package with pybind11 cpp extension☆57Updated 5 years ago
- Neural network verification in JAX☆144Updated 2 years ago
- Butterfly matrix multiplication in PyTorch☆178Updated 2 years ago
- Optimizing PAC-Bayes bounds for Stochastic Neural Networks with Gaussian weights☆28Updated 5 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 5 years ago
- ☆83Updated 5 years ago
- Distributed K-FAC preconditioner for PyTorch☆93Updated 2 weeks ago
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆248Updated 5 years ago
- PyTorch-SSO: Scalable Second-Order methods in PyTorch☆148Updated 2 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆129Updated 2 years ago
- ☆172Updated last year
- ☆133Updated 4 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆132Updated 6 years ago
- Integrate pytorch☆46Updated 4 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- ☆145Updated 2 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 3 years ago
- Structured matrices for compressing neural networks☆67Updated 2 years ago
- ☆33Updated 5 years ago