Programming for the Planet @propl.dev · Jun 15 1 / 4
Enzyme is useful in real climate problems like the shallow water equations. It saves both execution time and storage costs by smart checkpointing in the compiler pipeline and can run for large tasks. Scientists don't write TPU code typically, but code can be compiled using a parallel.for in Enzyme
1 likes 1 replies
?
Replies
Enzyme is full of optimisations like communications fusion and low precision arithmetic towards order of magnitude speedup when running on large deployments. Compilers are a promising route to avoid earth scientists to become domain scientists in high performance computing!