Affinity: Deep learning library for molecular geometry

https://affinity.mit.edu/