Use lightning 2.0-style device definitions in trainer
Fixes training on CPU-only installations. I haven't tested this as I'm a bit too lazy to set up a whole eScriptorium installation right now but it's functionally equivalent to the current kraken code.