Without proper operations "jitting" and correct device API calls, trainer is less than usable. - [x] Jit operations - [ ] Handle devices correctly
Without proper operations "jitting" and correct device API calls, trainer is less than usable.