Pyqg Jax | Anaconda.org

conda-forge / packages / pyqg-jax 0.8.1

0

Quasigeostrophic model in JAX (port of PyQG)

copied from cf-staging / pyqg-jax

License: MIT
Home: https://github.com/karlotness/pyqg-jax
Development: https://github.com/karlotness/pyqg-jax
Documentation: https://pyqg-jax.readthedocs.io/
4134 total downloads
Last upload: 9 months and 13 days ago

Installers

noarch v0.8.1

conda install

To install this package run one of the following:
conda install conda-forge::pyqg-jax

Description

This is a partial port of PyQG to JAX which enables GPU acceleration, batching, automatic differentiation, etc.

© 2024 Anaconda, Inc. All Rights Reserved. (v4.0.6) Legal | Privacy Policy