For CI and prod, we should have a CPU fallback. Currently, we need CUDA, which is not suitable in all cases.
For CI and prod, we should have a CPU fallback. Currently, we need CUDA, which is not suitable in all cases.