This is the last two weeks of my GSoC journey.

GSoC project

• GSoC project URL
• Work Repository

Tasks

1. Make a final polish to the code (branch):
   • Refactor some code in train_torch.py to make it more readable.
   • Remove unused commented-out blocks of code.
   • Add documentation for many functions.
2. Finalize all unit tests for the dataloader, ODE solver, util functions (loss functions), and training loop with pytest (branch).
3. Make a final polish to the README.md (branch).
4. Add a Quick Start Colab notebook on how to set up CellBox Pytorch and run a simple job.
5. Configure Github Actions to run on all unit tests when pushing to the branch.
6. Write a final report for wrapping up GSoC.

Relevant issues:

• clean-code branch
• final branch

Notes

Check out my personal GSoC notebook, where I keep my observations and comments in one place.