Would you consider adding basic unit tests or CI support?

[thesfy]

Even minimal tests would help maintain code quality, such as:

importing modules

running a forward pass

verifying dataloader behavior

small-scale training smoke test

CI support (GitHub Actions) would ensure future changes do not break functionality.