Combining Lightning with Brevitas for QAT

[jacopoabramo]

Greetings,

I'm fairly new to Lightning and I am considering using it for my projects for its close proximity with how it implements the training loop with Keras. One of my goals though is to train quantized models which I then try to deploy on FPGAs using external frameworks.

My question would be: can I implement a Lighitning module which can implement models created using the Brevitas framework, or any other framework for that matter? And if so, how? Is a Lightning module exported just as any normal model would be?

Thank you!