published Published 18 days ago
DIFFICE-jax: Differentiable neural-network solver for data assimilation of ice shelves in JAX
Python MATLAB Jupyter Notebook
published Published about 1 month ago
AutoPDEx: An Automized Partial Differential Equation solver based on JAX
Python
published Published about 1 month ago
jaxKAN: A unified JAX framework for Kolmogorov-Arnold Networks
Python
published Published about 2 months ago
Dynamax: A Python package for probabilistic state space modeling with JAX
Python
published Published about 2 months ago
unxt: A Python package for unit-aware computing with JAX
Python
published Published 3 months ago
jax-smfsb: A Python library for stochastic systems biology modelling and inference
Python
published Published 4 months ago
SBIAX: Density-estimation simulation-based inference in JAX
Python Roff
published Published 6 months ago