Akash Haridas issues

Repositories
Issues
Comments

Results 2 issues of


                                            Akash Haridas

Request: Example notebook showing training of a hybrid model

The readme says that an example notebook showing how to train a simple hybrid ML + CFD model is being prepared. Is there an ETA on this? I'm trying to...

Add ROCm support

* Add a Transformer Engine wrapper to enable calling Flash Attention on ROCm. * Add a config and launch script to demonstrate training Fuji-v2-70B on a single MI300 node.