patrick-kidger / wadler_lindig
A Wadler--Lindig pretty printer for Python
☆35Updated 3 weeks ago
Alternatives and similar repositories for wadler_lindig:
Users that are interested in wadler_lindig are comparing it to the libraries listed below
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago
- Stencil computations in JAX☆70Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆44Updated 2 months ago
- JAX Arrays for human consumption☆90Updated last year
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 3 months ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆18Updated 4 months ago
- ☆17Updated 2 months ago
- Exponential families for JAX☆63Updated this week
- A functional training loops library for JAX☆86Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- ☆59Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆47Updated 2 weeks ago
- ☆136Updated this week
- Pytorch-like dataloaders for JAX.