Going in I was really worried there would be a bunch of wonkiness with unique apis different from torch or tf. But it seems pretty close to torch, which is a pleasant surprise. No need to reinvent the wheel.