twitter / dict_minimize
Access scipy optimizers from your favorite deep learning framework.
☆77Updated last year
Related projects: ⓘ
- Documentation:☆119Updated last year
- ☆156Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Image augmentation library for Jax☆36Updated 5 months ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆57Updated 4 years ago
- Lightweight ML Experiment Logging 📖☆79Updated 3 weeks ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆41Updated 3 years ago
- A Pytree Module system for Deep Learning in JAX☆215Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆103Updated 2 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆50Updated 3 years ago
- A Python library for defining flat or nested dictionary flags.☆30Updated last year
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- ☆191Updated 4 months ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆40Updated 3 weeks ago
- ☆97Updated 2 months ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆76Updated 6 months ago
- Functional tensors for probabilistic programming☆235Updated last year
- Automate issue discovery for your projects against Lightning nightly and releases.☆45Updated last week
- Example Distill article repository—clone, rename, start writing!☆68Updated last year
- A modular system for machinable research code☆34Updated last month
- ☆149Updated 9 months ago
- Normalizing Flows using JAX☆82Updated 9 months ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated last year
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- A lightweight library for tensorflow 2.0☆66Updated 4 years ago
- A Python package of computer vision models for the Equinox ecosystem.☆100Updated 2 months ago
- Experiment orchestration☆103Updated 4 years ago
- ☆104Updated last week
- Normalizing Flows in Jax☆105Updated 4 years ago