published Published 12 days ago
AutoPDEx: An Automized Partial Differential Equation solver based on JAX
Python
published Published 16 days ago
jaxKAN: A unified JAX framework for Kolmogorov-Arnold Networks
Python
published Published 20 days ago
Dynamax: A Python package for probabilistic state space modeling with JAX
Python
published Published 3 months ago
jax-smfsb: A Python library for stochastic systems biology modelling and inference
Python
published Published 3 months ago
SBIAX: Density-estimation simulation-based inference in JAX
Python Roff
published Published 5 months ago
ReLax: Efficient and Scalable Recourse Explanation Benchmarking using JAX
Python Jupyter Notebook SCSS
published Published 8 months ago