mle-infrastructure / mle-logging
Lightweight ML Experiment Logging π
β80Updated 8 months ago
Alternatives and similar repositories for mle-logging:
Users that are interested in mle-logging are comparing it to the libraries listed below
- Lightweight Cluster/Cloud VM Job Management πβ41Updated 8 months ago
- Lightweight Experiment & Resource Monitoring πΊβ187Updated 2 years ago
- Lightweight Hyperparameter Optimization πβ147Updated 8 months ago
- A Pytree Module system for Deep Learning in JAXβ214Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.β46Updated 4 years ago
- Deep Probabilistic Programming Course @ DIKUβ57Updated 4 years ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training endsβ77Updated 5 months ago
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Cyclemoid implementation for PyTorchβ90Updated 3 years ago
- A Python package of computer vision models for the Equinox ecosystem.β103Updated 9 months ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"β180Updated 3 years ago
- β99Updated 3 years ago
- Image augmentation library for Jaxβ39Updated last year
- Gaussian Processes and Statistical Decision-making in Non-Euclidean Spacesβ198Updated 3 years ago
- A metrics library for the JAX ecosystemβ40Updated 2 years ago
- Official code for the Stochastic Polyak step-size optimizerβ139Updated 10 months ago
- WandB sweeps integration with Hydra sweeperβ48Updated last year
- My implementation of DeepMind's Perceiverβ63Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)β78Updated 4 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classiβ¦β129Updated 2 years ago
- β192Updated 3 weeks ago
- A collection of code snippets for my PyTorch Lightning projectsβ107Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β109Updated 2 years ago
- β114Updated this week
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)β42Updated 3 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributionsβ258Updated last year
- A library for uncertainty quantification based on PyTorchβ122Updated 3 years ago
- Functional deep learningβ108Updated 2 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.β77Updated last year
- β153Updated 4 years ago