7 projects
coordax
Coordinate axes for scientific computing in JAX
dinosaur
Differentiable Dynamics for Global Atmospheric Modeling 🦖
xarray-beam
Distributed Xarray with Apache Beam
neuralgcm
Neural General Circulation Models for Weather and Climate
jax-solar
Top of atmosphere incident solar radiation, in JAX
jax-cfd
None
tree-math
Mathematical operations for JAX pytrees
