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
Sorting:
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆29Updated 3 months ago
- ☆33Updated 4 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆109Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆47Updated 2 months ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- ☆37Updated 3 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated 2 years ago
- Distributed Bayesian Optimization☆23Updated 4 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- Deep learning with a multiplication budget☆47Updated 6 years ago
- Distributed K-FAC preconditioner for PyTorch☆86Updated this week
- Structured matrices for compressing neural networks☆66Updated last year
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- ☆33Updated 6 years ago
- Pytorch optimizers implementing Hilbert Constrained Gradient Descent☆19Updated 6 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- ☆15Updated 5 years ago
- Code for NeurIPS 2019 paper: "Symmetry-Based Disentangled Representation Learning requires Interaction with Environments" by H. Caselles-…☆35Updated 5 years ago
- ☆55Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Updated last year
- PyTorch implementation of HashedNets☆36Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."☆44Updated 6 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- Statistical adaptive stochastic optimization methods☆32Updated 5 years ago
- Hyperparameter tuning via uncertainty modeling☆47Updated last year