A JAX-based differentiable Lattice Boltzmann library for single and multiphase flow simulations on CPUs, GPUs and TPUs
No known vulnerabilities in the latest version
Based on latest version 0.5.7. If you're running an older version, check OSV for your specific version.
[](/packages/jax-lab)
<a href="/packages/jax-lab"><img src="/api/badges/jax-lab?period=month" alt="PyPI Stats"></a>