mlbench / mlbench-coreLinks
MLBench Framework Core Python Library
☆18Updated 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:
- PyTorch-MPI-DDP-example☆18Updated 7 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆109Updated 2 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Updated 4 years ago
- ☆71Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆37Updated 3 years ago
- ☆30Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- Limitations of the Empirical Fisher Approximation☆49Updated 11 months ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Updated 6 months ago
- ☆162Updated 2 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆61Updated 3 years ago
- Distributed K-FAC preconditioner for PyTorch☆94Updated last week
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆119Updated 3 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 6 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆63Updated 4 years ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆107Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated last year
- Code for NeurIPS 2019 paper: "Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processes…☆249Updated 5 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆129Updated last year
- Some small scale experiments for my blog posts 📝☆80Updated 3 years ago
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆21Updated 2 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- Neural network verification in JAX☆144Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- Integrate pytorch☆46Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆211Updated last year
- ☆33Updated 5 years ago