physicsnemo
physicsnemo copied to clipboard
⛰️[EPIC]: FIGConvNet integration for aerodynamic evaluation
### Integration
- [x] Integration of the TPUN
- [x] Enable Hydra config
- [ ] Add docstrings
- [ ] Add license headers
- [ ] Add README
- [ ] Use Modulus Distributed Manager
- [ ] Use Modulus checkpointing
- [ ] Use Modulus logging
### Verification
- [ ] Verify on the DrivAER dataset
- [ ] Verify on the Ahmed body dataset
### Optimization
- [ ] Enable AMP
- [ ] Enable Tensor Core compute
- [ ] Enable CUDA Graphs
- [ ] Enable JIT compilation
- [ ] Profile the code and identify the bottlenecks
- [ ] Other performance optimizations identified in profiles
### Maintenance
- [ ] Develop unit tests
- [ ] Perform QA
@mnabian What we please add the details to the scope of this epic?
@Alexey-Kamenev do you have another epic for FigConvNet integration? If so, we can close this one as duplicate.
No, that's the only one. I will rename it to use FIGConvNet instead of TPUN (that's the old name).