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
- Pretrained TorchVision models on CIFAR10 dataset (with weights)β24Updated 4 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.β44Updated 4 years ago
- π Pytorch code for the Nero optimiser.β20Updated 2 years ago
- Code release to reproduce ASHA experiments from "Random Search and Reproducibility for NAS."β22Updated 5 years ago
- Graph Convolutional Networks in JAXβ32Updated 4 years ago
- Notebooks for IPAM Tutorial, March 15 2019β24Updated 6 years ago
- Tensor Parallelism with JAX + Shard Mapβ11Updated last year
- notebooks of cool EBM visualizationsβ15Updated 4 years ago
- β36Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAXβ44Updated 3 years ago
- Code for fast dpsgd implementations in JAX/TFβ59Updated 2 years ago
- Public Codebase for Rethinking Parameter Counting: Effective Dimensionality Revisitedβ37Updated 2 years ago
- Official repository for our ICLR 2021 paper Evaluating the Disentanglement of Deep Generative Models with Manifold Topologyβ35Updated 4 years ago
- Code base for SRSGD.β28Updated 5 years ago
- Deterministic Decoding for Discrete Data in Variational Autoencodersβ24Updated 4 years ago
- β22Updated 6 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)β61Updated 3 years ago
- ImplΓ©mentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**β43Updated 5 years ago
- "Moshpit SGD: Communication-Efficient Decentralized Training on Heterogeneous Unreliable Devices", official implementationβ29Updated last month
- Collection of snippets for PyTorch usersβ25Updated 3 years ago
- Template repository for a Python 3-based data science project that uses Horovod.β43Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descentβ27Updated 5 months ago
- Isomorphisms of quiver representations applied to neural networks.β16Updated 8 months ago
- Hessian trace estimation using PyTorch and Hutch++β19Updated 4 years ago
- A minimal implementation of a VAE with BinConcrete (relaxed Bernoulli) latent distribution in TensorFlow.β21Updated 5 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suiteβ44Updated 3 years ago
- Inference on non-linear dynamical systems written in JAXβ11Updated 4 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
- Official code for UnICORNN (ICML 2021)β27Updated 3 years ago
- Hyperparameter tuning via uncertainty modelingβ47Updated 10 months ago