Machine learning code for use on the ATLAS Experiment