Apple Starflow is open source and requires Python, PyTorch, and a GPU, so you can try research builds and measure speed gains ...