Misc small improvements
- add logs/ to vs code exclude
- move some utils around
- don't set a strategy by default: lightning will infer this
- improvee messaging for missing input group
- various changes to try to support
torch.compile
see: #19 !127 (merged)