Metadata-Version: 2.4
Name: terrax
Version: 0.0.5
Summary: Terrax: Library for AI-first Earth System Modeling
Author-email: Google LLC <noreply@google.com>
License-Expression: Apache-2.0
Project-URL: source, https://github.com/neuralgcm/terrax
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: coordax>=0.1.5
Requires-Dist: dinosaur>=1.3.6
Requires-Dist: einops
Requires-Dist: etils
Requires-Dist: fiddle>=0.3.0
Requires-Dist: flax>=0.12.0
Requires-Dist: grain
Requires-Dist: jax
Requires-Dist: jax-datetime>=0.1.2
Requires-Dist: jax-solar
Requires-Dist: jaxlib
Requires-Dist: numpy
Requires-Dist: optax
Requires-Dist: orbax
Requires-Dist: pandas
Requires-Dist: tqdm
Requires-Dist: treescope
Requires-Dist: xarray>=2026.4.0
Requires-Dist: zarr>=3.1.4
Provides-Extra: tests
Requires-Dist: absl-py; extra == "tests"
Requires-Dist: chex; extra == "tests"
Requires-Dist: pytest; extra == "tests"
Dynamic: license-file

# Terrax: Library for AI-first Earth System Modeling

`Terrax` is a Python package for **building, training, and running** AI-first Earth System Models (ESMs) and their components. It is built on `jax`, `nnx`, `coordax`, and `xarray`, enabling efficient hardware acceleration, powerful automatic differentiation, and seamless integration with the open-source geospatial ecosystem.


## Contributing

See [`CONTRIBUTING.md`](CONTRIBUTING.md) for details.

## License

Apache 2.0; see [`LICENSE`](LICENSE) for details.

## Disclaimer

This project is not an official Google project. It is not supported by
Google and Google specifically disclaims all warranties as to its quality,
merchantability, or fitness for a particular purpose.
