AI Google Researchers Introduce An Open-Source Library in JAX for Deep Learning on Spherical Surfaces