Core Concepts

Lightning IR is built on top of PyTorch Lightning and shares most of its core concepts and API. Concretely, the LightningIRModule handles the model, the LightningIRDataModule handles the data, and the LightningIRTrainer handles combining the model and data. The Model, Data, and Trainer sections provide more details on these core concepts.