search

pytorch / opacus

Training PyTorch models with differential privacy
https://opacus.ai
Apache License 2.0
1.65k stars 328 forks source link

How to integrate Opacus with Nvidia NeMo #593

Closed yangjundy closed 11 months ago

yangjundy commented 1 year ago

NeMo is a framework that is developed based on PyTorch and PyTorch Lightning. It has been noticed that Opacus is compatible with PyTorch Lightning, which opens up the possibility of leveraging Opacus within the NeMo framework.

It would be incredibly advantageous if an integration between Opacus and NeMo could be achieved without making significant modifications to the existing NeMo modules, such as ASR.

Any suggestions or guidance on how to accomplish this seamless integration would be highly appreciated.

### Tasks
### Tasks
- [x] Feature improvements
yangjundy commented 11 months ago

I found that it works without much efforts.