Totom3

### Update It appears that any model which contains a `ReversibleLayer` (or subclass of) will produce such an error. The following code produces the same errors. The precise error depends...

No, I got it from pip. However in the end I managed to make it work by passing `use_memory_efficient_trainer=True` as an argument to `Loop`. (I'm not sure if this should...