mlbench / mlbench-core
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
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated 2 years ago
- ☆22Updated 6 years ago
- ☆33Updated 4 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆45Updated 5 years ago
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 6 years ago
- Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."☆43Updated 6 years ago
- Code base for SRSGD.☆28Updated 5 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆29Updated 2 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Code release to accompany paper "Geometry-Aware Gradient Algorithms for Neural Architecture Search."☆25Updated 4 years ago
- PyTorch-MPI-DDP-example☆17Updated 7 years ago
- Graph Convolutional Networks in JAX☆32Updated 4 years ago
- ☆74Updated 5 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- Distributed K-FAC Preconditioner for PyTorch☆85Updated this week
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- ☆19Updated 4 years ago
- ☆12Updated 4 years ago
- Structured matrices for compressing neural networks☆66Updated last year
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆47Updated 5 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- PyTorch implementation of HashedNets☆36Updated 2 years ago
- Butterfly matrix multiplication in PyTorch☆169Updated last year
- Proceedings of ICML 2019☆33Updated 2 years ago