twitter / dict_minimize
Access scipy optimizers from your favorite deep learning framework.
☆80Updated 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
- A Pytree Module system for Deep Learning in JAX☆213Updated last year
- Mixture Density Networks (Bishop, 1994) tutorial in JAX☆58Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- A Python library for defining flat or nested dictionary flags.☆31Updated last year
- Image augmentation library for Jax☆37Updated 9 months ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- ☆153Updated 4 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆47Updated 3 years ago
- Mathematical operations for JAX pytrees☆198Updated last month
- Evaluation Framework for Probabilistic Programming Languages☆99Updated last year
- Composable kernels for scikit-learn implemented in JAX.☆42Updated 4 years ago
- ☆100Updated 7 months ago
- Tools for JAX☆44Updated last week
- Functional tensors for probabilistic programming☆237Updated last year
- ☆153Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- ☆108Updated 2 years ago
- Hessian spectral density estimation in TF and Jax☆120Updated 4 years ago
- ☆73Updated this week
- Jax SSM Library☆49Updated 2 years ago
- Annotating tensor shapes using Python types☆158Updated last year
- Example Distill article repository—clone, rename, start writing!☆69Updated 2 years ago
- Recognizing and exploiting conjugacy without a domain-specific language☆36Updated 5 years ago
- Collection of eclectic utils for python.☆209Updated this week
- ☆191Updated 8 months ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆41Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- Lightweight ML Experiment Logging 📖☆78Updated 5 months ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆45Updated this week