The full training script for Enformer - Tensorflow Sonnet - View it on GitHub