Roadmap for SimulAI

[Joao-L-S-Almeida]

Short term development (fundamental steps) :

• Coverage for more than 95 % of the source code.
• Complete documentation for all the classes and methods.
• Automating the execution of sanity tests and coverage.
• Examples in notebooks for all the cases seen in the directory examples.

Medium term Development

• Complete scalability tests for Data Distributed Parallelism (DDP) using SimulAI models.
• Complete scalability tests Full-Sharded Data Parallelism (FSDP) using SimulAI models.
• Pipelines execution optimized via JIT (TorchScript, maybe PyPy).
• Integration with the torchvision Modules ViTs, U-Nets and others.

To be considered

• There can be gain in converting some modules to Cython ?