Machine learning (ML) of unresolved processes offers many new possibilities for improving weather and climate models, but integrating ML into the models has been an engineering challenge, and there are performance issues. We present a new software plugin for this integration, TorchClim, that is scalable and flexible and thereby allows a new level of experimentation with the ML approach. We also provide guidance on ML training and demonstrate a skillful hybrid ML atmosphere model.