twitter / dict_minimizeLinks
Access scipy optimizers from your favorite deep learning framework.
☆79Updated 2 years ago
Alternatives and similar repositories for dict_minimize
Users that are interested in dict_minimize are comparing it to the libraries listed below
Sorting:
- Documentation:☆120Updated 2 years ago
- Evaluation Framework for Probabilistic Programming Languages☆100Updated last year
- Experiment orchestration☆103Updated 5 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- A simple wrapper over `pydot` and `graphviz` which fixes some sharp edges☆63Updated 2 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆60Updated 5 years ago
- Example Distill article repository—clone, rename, start writing!☆69Updated 2 years ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- Functional machine learning for fun☆85Updated 4 years ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- ☆153Updated 5 years ago
- Repository of models in Pyro☆29Updated 11 months ago
- Automatic differentiation + optimization☆103Updated 6 years ago
- Image augmentation library for Jax☆39Updated last year
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- A Python library for defining structured command-line flags.☆31Updated 2 weeks ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A queue service for quickly developing scripts that use all your GPUs efficiently☆85Updated 2 years ago
- ☆30Updated 3 years ago
- Blazingly fast capsule networks in 75 lines of pytorch+einops☆26Updated 3 years ago
- A python library for stochastic variational inference and differentiable probabilistic programming☆33Updated 6 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 5 years ago
- ☆108Updated 2 years ago
- ☆192Updated 3 weeks ago
- Functional tensors for probabilistic programming☆241Updated last year
- ☆76Updated last week
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago