Improve initial data loading in Ahmed body example.

[Alexey-Kamenev]

Modulus Pull Request

Description

Improvements to Ahmed body dataset and example:

• improved initial dataset loading time.
• minor refactoring to the dataset
• added an option to control checkpoint frequency.

Checklist

• [x] I am familiar with the Contributing Guidelines.
• [x] New or existing tests cover these changes.
• [x] The documentation is up to date with these changes.
• [ ] The CHANGELOG.md is up to date with these changes.
• [ ] An issue is linked to this pull request.

Dependencies

[Alexey-Kamenev]

/blossom-ci

[Alexey-Kamenev]

/blossom-ci