using torch.compile over sfast for compatibility; other smol changes 05e29c3 rynmurdock commited on May 2