Skip to content

Release AlphaNet v0.1.1

Choose a tag to compare

View all tags
@zmyybc zmyybc released this 03 Sep 12:11
· 31 commits to jax_and_zbl since this release
v0.1.1
b00645d

Major Changes

  1. Jax version to speed up

    • Provide 3x speedup AlphaNet written in JAX, with a FLAX version and HaiKu version.
    • Users can convert the checkpoint trained in torch to our JAX model, we are the only one providing this

  2. Added tuned Ziegler–Biersack–Littmark (ZBL) potential to stablize AlphaNet

    • We followed the ZBL theory, and fitted our own ZBL potential by calculated ~7,000 diatom systems.
Assets 2
Loading