But then you would not be able to run it on GPU, nor compile to XLA for a ~5x speedup, nor use it inside a TensorFlow codebase or a PyTorch codebase.
GPU acceleration and XLA compilation limitations in framework integration
By
–
By
–
But then you would not be able to run it on GPU, nor compile to XLA for a ~5x speedup, nor use it inside a TensorFlow codebase or a PyTorch codebase.