mle-infrastructure / mle-logging
Lightweight ML Experiment Logging π
β80Updated 7 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 7 months ago
- Lightweight Hyperparameter Optimization πβ146Updated 7 months ago
- Lightweight Experiment & Resource Monitoring πΊβ187Updated 2 years ago
- A Pytree Module system for Deep Learning in JAXβ214Updated 2 years ago
- Image augmentation library for Jaxβ39Updated last year
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.β46Updated 3 years ago
- A collection of code snippets for my PyTorch Lightning projectsβ107Updated 4 years ago
- Cyclemoid implementation for PyTorchβ89Updated 3 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorchβ88Updated 10 months ago
- A Python package of computer vision models for the Equinox ecosystem.β103Updated 8 months ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training endsβ77Updated 4 months ago
- A library for uncertainty quantification based on PyTorchβ123Updated 3 years ago
- β99Updated 3 years ago
- Check if you have training samples in your test setβ64Updated 2 years ago
- My implementation of DeepMind's Perceiverβ63Updated 3 years ago
- Running Jax in PyTorch Lightningβ93Updated 3 months ago
- WandB sweeps integration with Hydra sweeperβ47Updated last year
- A metrics library for the JAX ecosystemβ40Updated 2 years ago
- π§ Pytorch code for the Fromage optimiser.β123Updated 8 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β108Updated 2 years ago
- Efficiently Composable Data Augmentation on the GPU with Jaxβ33Updated 8 months ago
- Normalizing Flows using JAXβ83Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.β77Updated last year
- Distributed skorch on Ray Trainβ57Updated 2 years ago
- Unofficial JAX implementations of deep learning research papersβ154Updated 2 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
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"β180Updated 3 years ago
- β192Updated 10 months ago
- A library for evaluating representations.β76Updated 3 years ago