tree-math makes it easy to implement numerical algorithms that work on JAX pytrees, such as iterative methods for optimization and equation solving. It does so by providing a wrapper class ...

What began with a focus on weather forecasting has evolved toward addressing errors in scientific modeling. In the collaborative environment of the Penn State Institute for Computational and Data ...