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
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- 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
- ☆67Updated last year
- A functional training loops library for JAX☆85Updated 9 months ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- ☆97Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- WandB sweeps integration with Hydra sweeper☆46Updated 9 months ago
- A Pytree Module system for Deep Learning in JAX☆214Updated last year
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- ☆33Updated last year
- ☆105Updated 2 weeks ago
- ☆47Updated 2 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Deep Probabilistic Programming Course @ DIKU☆56Updated 4 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 4 months ago
- Riemannian Convex Potential Maps☆68Updated last year
- Codebase for Learning Invariances in Neural Networks☆94Updated 2 years ago
- ☆6Updated last year
- Running Jax in PyTorch Lightning☆82Updated 2 weeks ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- A library for evaluating representations.☆76Updated 3 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆105Updated last year
- ☆155Updated 4 years ago
- Lightweight Hyperparameter Optimization 🚂☆145Updated 2 months ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 3 years ago