Thanks for the insights. I think you may want to try using MLX instead of PyTorch or TF, but you can only use it on Mac.