What's Changed
- Enable swa if start_swa was given by @vue1999 in #697
- Update foundations models to contain 0b by @alinelena in #691
- extract_invariant now correctly extracts first layer when num_layers=1 by @RokasEl in #695
- Add cuequivariance support by @ilyes319 in #704
- Add cuequivariance support by @ilyes319 in #709
- Fix-extract-equivariant-features-with-num-layers-1 by @RokasEl in #703
- add stress to the cueq test by @ilyes319 in #712
- add option pt_train_file == "mp" by @ilyes319 in #716
- bum version by @ilyes319 in #719
Full Changelog: v0.3.8...v0.3.9