Any mace-torch that is compiled with latest e3nn>=0.5.7?

[sainyamn]

Hello @ilyes319,
I was wanting to use intermediate layers from MACE-MPA-0 (frozen) and decompose them into individual irreps for creating custom heads on top for full dielectric tensor predictions, I wanted to use the latest version of e3nn which gives an option to filter the irreps directly. I see all the mace-torch have used e3nn 0.4.4. Any plans on upgrading that to the latest?
Aside, do you have any suggestions on which pretrained mace could be best for this task even though its not directly trained on polarization data?
Thanks!