The always prolific Lucidrains has been implementing AlphaFold3 in PyTorch:
https://github.com/lucidrains/alphafold3-pytorch
This illustrated explainer is a great breakdown of the model in general:
https://elanapearl.github.io/blog/2024/the-illustrated-alphafold/