mlbench / mlbench-coreLinks
MLBench Framework Core Python Library
☆17Updated 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:
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Graph Convolutional Networks in JAX☆32Updated 4 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆29Updated 4 months ago
- PyTorch-MPI-DDP-example☆17Updated 7 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Pytorch optimizers implementing Hilbert Constrained Gradient Descent☆19Updated 6 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Distributed Bayesian Optimization☆23Updated 4 years ago
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- ☆15Updated 5 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Template repository for a Python 3-based data science project that uses Horovod.☆43Updated 3 years ago
- PyTorch implementation of HashedNets☆36Updated 2 years ago
- ☆22Updated 7 years ago
- ☆55Updated last year
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated 2 years ago
- Optimizing PAC-Bayes bounds for Stochastic Neural Networks with Gaussian weights☆27Updated 4 years ago
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 6 years ago
- Deep learning with a multiplication budget☆47Updated 6 years ago
- A differentiation API for PyTorch☆30Updated 4 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆59Updated 3 years ago
- Distributed K-FAC preconditioner for PyTorch☆87Updated last week
- Statistical adaptive stochastic optimization methods☆32Updated 5 years ago
- A custom PyTorch layer that is capable of implementing extremely wide and sparse linear layers efficiently☆49Updated last year
- ☆37Updated 3 years ago
- ☆74Updated 6 years ago