【cuEquivariance：NVIDIA开发的高性能神经网络库，帮助构建尊重对称性的神经网络模型，提升模型的数据效率】cuEquivariance is an NVIDIA Python library designed to facilitate the construction of high-performance equivariant neural networks using segmented tensor products. cuEquivariance provides a comprehensive API for describing segmented tensor products and optimized CUDA kernels for their execution. Additionally, cuEquivariance offers bindings for both PyTorch and JAX, ensuring broad compatibility and ease of integration. GitHub: github.com/NVIDIA/cuEquivariance #神经网络# #CUDA# #机器学习#
|
