This project implements a complete pipeline for protein function prediction based on "Deep Learning for Biologists" Chapter 2, converted from JAX/Flax to PyTorch.