Rocketknight1 HF Staff
Rename package esmfold2_trimul -> esmfold2_trimul_kernel to match repo-derived package name
5be2771 verified | from .layers import ESMFold2TriangleMultiplication | |
| from .trimul_with_residual import triangle_multiplicative_update_with_residual | |
| __all__ = [ | |
| "ESMFold2TriangleMultiplication", | |
| "triangle_multiplicative_update_with_residual", | |
| ] | |