published Published about 1 month ago
DiRe - JAX: A JAX based Dimensionality Reduction Algorithm for Large-scale Data
Python
published Published 3 months ago
DIFFICE-jax: Differentiable neural-network solver for data assimilation of ice shelves in JAX
Python MATLAB Jupyter Notebook
published Published 3 months ago
AutoPDEx: An Automized Partial Differential Equation solver based on JAX
Python
published Published 3 months ago
jaxKAN: A unified JAX framework for Kolmogorov-Arnold Networks
Python
published Published 4 months ago
Dynamax: A Python package for probabilistic state space modeling with JAX
Python
published Published 5 months ago
jax-smfsb: A Python library for stochastic systems biology modelling and inference
Python
published Published 6 months ago