The Open MatSci ML Toolkit: A Flexible Framework for Machine Learning in Materials Science