GitHub repository: GridTools/jace
Primary language: Python
Description: JaCe - JAX Just-In-Time compilation using DaCe (Data Centric Parallel Programming)