mlbench / mlbench-core
MLBench Framework Core Python Library
☆16Updated last year
Alternatives and similar repositories for mlbench-core:
Users that are interested in mlbench-core are comparing it to the libraries listed below
- ☆22Updated 6 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆29Updated 2 weeks ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- A learning rate recommending and benchmarking tool.☆19Updated last year
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- ☆33Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- BrainProp: How the brain can implement reward-based error backpropagation☆17Updated 2 years ago
- ☆19Updated 3 years ago
- ☆36Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Pytorch optimizers implementing Hilbert Constrained Gradient Descent☆19Updated 5 years ago
- Collection of snippets for PyTorch users☆25Updated 2 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- Benchmarking PyTorch 2.0 different models☆21Updated last year
- PyTorch implementation of HashedNets☆36Updated last year
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆43Updated 7 months ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.☆21Updated 5 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated last week
- Tree Approximate Message Passing☆30Updated 9 months ago
- Proceedings of ICML 2019☆33Updated 2 years ago
- Deep learning with a multiplication budget☆47Updated 6 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated last year