twitter / dict_minimize
Access scipy optimizers from your favorite deep learning framework.
☆79Updated last year
Alternatives and similar repositories for dict_minimize:
Users that are interested in dict_minimize are comparing it to the libraries listed below
- Documentation:☆119Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆59Updated 5 years ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago
- ☆153Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- Evaluation Framework for Probabilistic Programming Languages☆100Updated last year
- Repository of models in Pyro☆29Updated 9 months ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Functional tensors for probabilistic programming☆240Updated last year
- Automatic differentiation + optimization☆103Updated 6 years ago
- A python library for stochastic variational inference and differentiable probabilistic programming☆32Updated 5 years ago
- Mathematical operations for JAX pytrees☆199Updated 5 months ago
- ☆30Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated last year
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- A modular system for machinable research code☆35Updated 3 weeks ago
- Jax SSM Library☆49Updated 2 years ago
- Recognizing and exploiting conjugacy without a domain-specific language☆36Updated 5 years ago
- Inference Combinators in JAX☆47Updated 6 months ago
- An extension to Sacred for automated hyperparameter optimization.☆59Updated 7 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Bayesian Bandits☆68Updated last year
- ☆103Updated 10 months ago
- ☆80Updated 3 years ago
- ☆36Updated 2 years ago