search

NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
934 stars 219 forks source link

⛰️[EPIC]: FIGConvNet integration for aerodynamic evaluation #354

Open mnabian opened 7 months ago

mnabian commented 7 months ago 
### 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
ram-cherukuri commented 3 weeks ago

@mnabian What we please add the details to the scope of this epic?

mnabian commented 2 weeks ago

@Alexey-Kamenev do you have another epic for FigConvNet integration? If so, we can close this one as duplicate.

Alexey-Kamenev commented 1 week ago

No, that's the only one. I will rename it to use FIGConvNet instead of TPUN (that's the old name).