mle-infrastructure / mle-scheduler
Lightweight Cluster/Cloud VM Job Management 🚀
☆41Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for mle-scheduler
- Lightweight ML Experiment Logging 📖☆80Updated 2 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Image augmentation library for Jax☆37Updated 7 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- ☆66Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆122Updated 3 months ago
- ☆97Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆58Updated last year
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- ☆104Updated this week
- A metrics library for the JAX ecosystem☆39Updated last year
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- A functional training loops library for JAX☆85Updated 8 months ago
- ☆156Updated 4 years ago
- ☆33Updated last year
- Pretrained models for Jax/Haiku; MobileNet, ResNet, VGG, Xception.☆23Updated 2 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 3 months ago
- Easy Hypernetworks in Pytorch and Jax☆95Updated last year
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- JAX Arrays for human consumption☆88Updated last year
- Official implementation of the paper "Topographic VAEs learn Equivariant Capsules"☆77Updated 2 years ago
- ☆98Updated 4 months ago
- My implementation of DeepMind's Perceiver☆63Updated 3 years ago
- ☆191Updated 5 months ago
- Clockwork VAEs in JAX/Flax☆31Updated 3 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- ☆78Updated 3 years ago
- ☆152Updated 10 months ago