谷歌研究人员最近推出了一个基于 JAX 的开源库，旨在解决在球面上进行深度学习的挑战。该库利用球面卷积和交叉相关操作处理球面数据，具有巨大潜力在分子性质预测和气象预测等领域。通过处理这些数据，研究人员能够提高模型的性能，解决数据采样和旋转稳健性的问题。这个开源库的发布标志着深度学习领域在处理复杂球面数据方面的一项重要进展。