mlbench / mlbench-core
MLBench Framework Core Python Library
☆16Updated 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
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 2 years ago
- Inference on non-linear dynamical systems written in JAX☆11Updated 4 years ago
- A lightweight library for tensorflow 2.0☆66Updated 5 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆20Updated last year
- A differentiation API for PyTorch☆30Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Code for fast dpsgd implementations in JAX/TF☆59Updated 2 years ago
- PyTorch-MPI-DDP-example☆17Updated 6 years ago
- ☆33Updated 4 years ago
- Statistical adaptive stochastic optimization methods☆33Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆47Updated last week
- Notebooks for IPAM Tutorial, March 15 2019☆24Updated 5 years ago
- ☆36Updated 3 years ago
- ☆55Updated last year
- ☆22Updated 6 years ago
- 👑 Pytorch code for the Nero optimiser.☆20Updated 2 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 3 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topology☆35Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Computing the eigenvalues of Neural Tangent Kernel and Conjugate Kernel (aka NNGP kernel) over the boolean cube☆48Updated 5 years ago
- Code for "Exponential Family Estimation via Adversarial Dynamics Embedding" (NeurIPS 2019)☆13Updated 5 years ago
- Experiments for Meta-Learning Symmetries by Reparameterization☆56Updated 3 years ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆44Updated 5 years ago
- Code for the CVPR 2021 paper: Understanding Failures of Deep Networks via Robust Feature Extraction☆35Updated 2 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated last year
- Graph Convolutional Networks in JAX☆32Updated 4 years ago