mlbench / mlbench-core
MLBench Framework Core Python Library
☆16Updated last year
Related projects ⓘ
Alternatives and complementary repositories for mlbench-core
- A learning rate recommending and benchmarking tool.☆19Updated last year
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."☆22Updated 5 years ago
- ☆22Updated 6 years ago
- Hessian trace estimation using PyTorch and Hutch++☆19Updated 4 years ago
- Collection of snippets for PyTorch users☆26Updated 2 years ago
- Code base for SRSGD.☆28Updated 4 years ago
- Integrate pytorch☆47Updated 3 years ago
- Inference on non-linear dynamical systems written in JAX☆12Updated 4 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementation☆28Updated last year
- Graph Convolutional Networks in JAX☆31Updated 3 years ago
- Personal solutions to the Triton Puzzles☆16Updated 4 months ago
- Repo to accompany paper "Implicit Self-Regularization in Deep Neural Networks..."☆41Updated 5 years ago
- Distributed K-FAC Preconditioner for PyTorch☆80Updated this week
- ☆36Updated 2 years ago
- Quickest way to share everything about your research within a single app☆17Updated 9 months ago
- ☆17Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisited☆36Updated last year
- PyTorch implementation of HashedNets☆36Updated last year
- ☆11Updated 4 years ago
- PyTorch implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆33Updated 2 years ago
- ☆52Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Updated last year
- Code for the CVPR 2021 paper: Understanding Failures of Deep Networks via Robust Feature Extraction☆35Updated 2 years ago
- Hyperparameter tuning via uncertainty modeling☆46Updated 6 months ago
- Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch☆87Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- ☆14Updated 2 years ago
- A LinearOperator implementation for PyTorch☆18Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 3 years ago