New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

sort instructions in tp_out_irreps_with_instructions #51

Merged

ilyes319 merged 1 commit into ACEsuit:torchscript_develop from mariogeiger:torchscript_develop

Nov 21, 2022

Contributor

mariogeiger commented Nov 21, 2022 •

edited

Loading

Reordering the interactions in the same order as the outputs simplifies the structure of the weights.

This help to convert the weights from torch to jax.


          sort instructions in tp_out_irreps_with_instructions

da0645d

mariogeiger changed the base branch from main to torchscript_develop

November 21, 2022 06:19

ilyes319 merged commit 7691341 into ACEsuit:torchscript_develop

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet