EquiTriton 是一个专为等变神经网络设计的高性能神经网络核工具,旨在提升神经网络的计算效率。它使用 Triton-lang 编写,提供了高性能的计算核,能够显著加速等变神经网络的训练和推理过程。该工具还支持在深度学习框架中集成高性能计算核,优化神经网络模型的推理速度,并支持大规模神经网络的并行计算。