bfloat16 version of Diffdock

gcorso / DiffDock

Implementation of DiffDock: Diffusion Steps, Twists, and Turns for Molecular Docking

https://arxiv.org/abs/2210.01776

MIT License

976 stars 238 forks source link

Open srilekha1993 opened 3 weeks ago

srilekha1993 commented 3 weeks ago

Hi, As the code is written in float 32, can you please update the bfloat16 version of inference.py run