Data

Tools

Indexes

Applications

Experiments

Open Source Science

dinosaur

NeuralGCM's differentiable dycore

https://github.com/neuralgcm/dinosaur

Science Score: 36.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ✓
    Academic publication links
    Links to: nature.com
  • ○
    Committers with academic emails
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (10.3%) to scientific vocabulary

Keywords

neuralgcm

Keywords from Contributors

deep-neural-networks distributed jax
Last synced: 10 months ago · JSON representation

Repository

NeuralGCM's differentiable dycore

Basic Info
  • Host: GitHub
  • Owner: neuralgcm
  • License: apache-2.0
  • Language: Jupyter Notebook
  • Default Branch: main
  • Homepage:
  • Size: 4.65 MB
Statistics
  • Stars: 294
  • Watchers: 6
  • Forks: 26
  • Open Issues: 3
  • Releases: 0
Topics
neuralgcm
Created over 2 years ago · Last pushed 12 months ago
Metadata Files
Readme Contributing License

README.md

Dinosaur: Differentiable Dynamics for Global Atmospheric Modeling 🦖

Authors: Jamie A. Smith, Dmitrii Kochkov, Peter Norgaard, Janni Yuval, Stephan Hoyer

Dinosaur is an old-fashioned (some might say prehistoric) dynamical core for global atmospheric modeling, re-written in JAX to meet the needs of modern AI weather/climate models:

  • Dynamics: Dinosaur uses spectral methods to solve the shallow water equations and the primitive equations (moist and dry) on sigma coordinates.
  • Auto-diff: Dinosaur supports both forward- and backward-mode automatic differentiation in JAX. This enables "online training" of hybrid AI/physics models.
  • Acceleration: Dinosaur is designed to run efficiently on modern accelerator hardware (GPU/TPU), including parallelization across multiple devices.

For more details, see our paper Neural General Circulation Models for Weather and Climate.

Usage instructions

Dinosaur is an experimental research project that we are still working on documenting.

We currently have three notebooks illustrating how to use Dinosaur:

We recommend running them using Google Colab with a GPU runtime. You can also install Dinosaur locally: pip install dinosaur

See also

If you like Dinosaur, you might also like SpeedyWeather.jl, which solves similar equations in Julia.

Contributing

See CONTRIBUTING.md for details. We are open to user contributions, but please reach out (either on GitHub or by email) to coordinate before starting significant work.

License

Apache 2.0; see LICENSE for details.

Owner

  • Name: neuralgcm
  • Login: neuralgcm
  • Kind: organization

GitHub Events

Total
  • Create event: 43
  • Issues event: 4
  • Release event: 9
  • Watch event: 63
  • Delete event: 27
  • Issue comment event: 8
  • Push event: 132
  • Pull request event: 68
  • Fork event: 9
Last Year
  • Create event: 43
  • Issues event: 4
  • Release event: 9
  • Watch event: 63
  • Delete event: 27
  • Issue comment event: 8
  • Push event: 132
  • Pull request event: 68
  • Fork event: 9

Committers

Last synced: 10 months ago

All Time
  • Total Commits: 76
  • Total Committers: 7
  • Avg Commits per committer: 10.857
  • Development Distribution Score (DDS): 0.329
Past Year
  • Commits: 34
  • Committers: 3
  • Avg Commits per committer: 11.333
  • Development Distribution Score (DDS): 0.294
Top Committers
Name Email Commits
Stephan Hoyer s****r@g****m 51
Dmitrii Kochkov d****v@g****m 18
Dinosaur authors n****y@g****m 2
Jake VanderPlas v****s@g****m 2
Ian Langmore l****e@g****m 1
Janni Yuval j****l@g****m 1
Rebecca Chen r****n@g****m 1
Committer Domains (Top 20 + Academic)
google.com: 7

Issues and Pull Requests

Last synced: 10 months ago

All Time
  • Total issues: 6
  • Total pull requests: 118
  • Average time to close issues: 13 days
  • Average time to close pull requests: 1 day
  • Total issue authors: 4
  • Total pull request authors: 4
  • Average comments per issue: 2.33
  • Average comments per pull request: 0.07
  • Merged pull requests: 57
  • Bot issues: 0
  • Bot pull requests: 113
Past Year
  • Issues: 5
  • Pull requests: 69
  • Average time to close issues: 13 days
  • Average time to close pull requests: 1 day
  • Issue authors: 3
  • Pull request authors: 2
  • Average comments per issue: 2.2
  • Average comments per pull request: 0.03
  • Merged pull requests: 36
  • Bot issues: 0
  • Bot pull requests: 67
View more stats
Top Authors
Issue Authors
  • whpy (2)
  • milankl (1)
  • sit23 (1)
  • shoyer (1)
Pull Request Authors
  • copybara-service[bot] (112)
  • jeromebarre (2)
  • shoyer (2)
  • deven367 (1)
Top Labels
Issue Labels
documentation (1) enhancement (1) help wanted (1)
Pull Request Labels

Packages

  • Total packages: 1
  • Total downloads:
    • pypi 5,161 last-month
  • Total dependent packages: 0
  • Total dependent repositories: 1
  • Total versions: 7
  • Total maintainers: 1
pypi.org: dinosaur

Differentiable Dynamics for Global Atmospheric Modeling 🦖

  • Versions: 7
  • Dependent Packages: 0
  • Dependent Repositories: 1
  • Downloads: 5,161 Last month
Rankings
Dependent packages count: 10.1%
Dependent repos count: 21.6%
Average: 27.4%
Downloads: 50.6%
Maintainers (1)
gauravghosh
Last synced: 10 months ago

Dependencies

.github/workflows/ci-build.yml actions
  • actions/cache v2 composite
  • actions/checkout v2 composite
  • actions/setup-python v2 composite
  • styfle/cancel-workflow-action 0.7.0 composite
pyproject.toml pypi
  • fsspec *
  • jax *
  • jaxlib *
  • numpy *
  • pandas *
  • pint *
  • scikit-learn *
  • scipy *
  • tree-math *
  • xarray *
  • xarray-tensorstore *