AdityaLab / GradABM
[AAMAS 2023] Differentiable Agent-based Epidemiology
☆14Updated 5 months ago
Related projects: ⓘ
- JAX Arrays for human consumption☆87Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆13Updated this week
- Einsum-like high-level array sharding API for JAX☆31Updated 2 months ago
- Bayesian inference with Python and Jax.☆31Updated last year
- Scalable Convex Neural Networks☆20Updated 4 months ago
- ☆46Updated this week
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆66Updated 2 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆39Updated 11 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆59Updated last year
- Agustinus' very opiniated publication-ready plotting library☆51Updated 2 weeks ago
- Graph neural networks in JAX.☆66Updated 3 months ago
- Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.☆58Updated last year
- An Open-Ended Agentic Simulator☆17Updated last month
- Loopy belief propagation for factor graphs on discrete variables in JAX☆124Updated 3 months ago
- A functional training loops library for JAX☆84Updated 7 months ago
- ☆85Updated this week
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆30Updated last month
- Minimal Implementation of Bayesian Optimization in JAX☆82Updated 4 months ago
- ☆18Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- probabilistic programming focused on fun☆21Updated last week
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Pytorch-like dataloaders in JAX.☆52Updated last month
- Public code for running Stochastic Gradient Descent on GPs.☆31Updated 5 months ago
- Exponential families for JAX☆54Updated last week
- ☆17Updated 3 months ago
- ☆34Updated 2 years ago
- Scaling scaling laws with board games.☆36Updated last year
- ☆56Updated last month