pyqg-jax
Quasigeostrophic model in JAX (port of PyQG)
Installation
To install this package, run one of the following:
Usage Tracking
0.8.1
0.8.0
0.7.0
0.6.0
4 / 8 versions selected
Downloads (Last 6 months): 0
Description
This is a partial port of PyQG to JAX which enables GPU acceleration, batching, automatic differentiation, etc.
About
Summary
Quasigeostrophic model in JAX (port of PyQG)
Last Updated
Feb 28, 2024 at 03:12
License
MIT
Total Downloads
9.5K
Version Downloads
2.6K
Supported Platforms
noarch
GitHub Repository
https://github.com/karlotness/pyqg-jaxDocumentation
https://pyqg-jax.readthedocs.io/