PyTorch Implementation of a Simplified AlphaFold2 for Spring 2022 CSCI6969 at RPI
https://magazine.rpi.edu/sharing-your-building-blocks
https://www.nature.com/articles/s41586-021-03819-2
- Training is done through
src/train.py
, along with parameter configuration - See the supplementary information document for parameter information and implementation details
- Code was tested on a very scaled down version of the complete model, giving less than optimal results
- Special thanks to einops and einsum
- TODO: Half-Precision Training, Complete Model Train/Test