Alternatives and similar repositories for mlbench-core

Users that are interested in mlbench-core are comparing it to the libraries listed below

• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• fsschneider / DeepOBS
  DeepOBS: A Deep Learning Optimizer Benchmark Suite
  ☆109Updated 2 years ago
• shreyansh26 / ML-Optimizers-JAX
  Toy implementations of some popular ML optimizers using Python/JAX
  ☆44Updated 4 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆45Updated 6 months ago
• HazyResearch / structured-nets
  Structured matrices for compressing neural networks
  ☆67Updated 2 years ago
• cybertronai / autograd-lib
  ☆33Updated 5 years ago
• NVIDIA / dllogger
  A logging tool for deep learning.
  ☆65Updated 10 months ago
• ayulockin / LossLandscape
  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
• gpauloski / kfac-pytorch
  Distributed K-FAC preconditioner for PyTorch
  ☆94Updated last week
• NERSC / slurm-ray-cluster
  ☆62Updated 2 years ago
• fKunstner / limitations-empirical-fisher
  Limitations of the Empirical Fisher Approximation
  ☆49Updated 11 months ago
• g-benton / hessian-eff-dim
  Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited
  ☆37Updated 3 years ago
• xhzhao / PyTorch-MPI-DDP-example
  PyTorch-MPI-DDP-example
  ☆18Updated 7 years ago
• TheSalon / fast-dpsgd
  Code for fast dpsgd implementations in JAX/TF
  ☆59Updated 3 years ago
• evcu / pytorchpruner
  ☆22Updated 7 years ago
• cybertronai / imagenet18
  Train ImageNet in 18 minutes on AWS
  ☆134Updated last year
• jxbz / fromage
  🧀 Pytorch code for the Fromage optimiser.
  ☆129Updated last year
• n2cholas / shapecheck
  Framework-agnostic library for checking array/tensor shapes at runtime.
  ☆46Updated 4 years ago
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated 2 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 4 years ago
• yandex-research / moshpit-sgd
  "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation
  ☆30Updated last year
• rgrosse / csc2541_examples
  ☆30Updated 4 years ago
• HazyResearch / butterfly
  Butterfly matrix multiplication in PyTorch
  ☆178Updated 2 years ago
• thegregyang / NTK4A
  Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"
  ☆107Updated 5 years ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated 2 years ago
• Distributed-AI / PipeTransformer
  PipeTransformer: Automated Elastic Pipelining for Distributed Training of Large-scale Models. ICML 2021
  ☆55Updated 4 years ago
• teddykoker / grokking
  PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆38Updated 4 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Updated 3 years ago
• SirRob1997 / Crowded-Valley---Results
  This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"
  ☆184Updated 4 years ago
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆133Updated 6 years ago