mle-infrastructure / mle-scheduler
Lightweight Cluster/Cloud VM Job Management 🚀
☆39Updated 5 months ago
Alternatives and similar repositories for mle-scheduler:
Users that are interested in mle-scheduler are comparing it to the libraries listed below
- Lightweight ML Experiment Logging 📖☆78Updated 5 months ago
- Very deep VAEs in JAX/Flax☆46Updated 3 years ago
- ☆67Updated last year
- Image augmentation library for Jax☆37Updated 9 months ago
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- ☆153Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- ☆98Updated 3 years ago
- ☆47Updated 2 years ago
- ☆100Updated 7 months ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A functional training loops library for JAX☆86Updated 11 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Normalizing Flows in Jax☆106Updated 4 years ago
- A library for evaluating representations.☆74Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 6 months ago
- ☆40Updated 2 months ago
- WandB sweeps integration with Hydra sweeper☆47Updated last year
- JAX Arrays for human consumption☆90Updated last year
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- ☆4Updated last year
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- A metrics library for the JAX ecosystem☆39Updated last year
- Clockwork VAEs in JAX/Flax☆32Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago