Relax cuda XLA_TARGET
jonatanklosko opened this issue · comments
Instead of cuda118 and cuda120 we can just do cuda11 and cuda12, which is what jaxlib does now. Then we just list what are the minimal supported versions for each.
Pre-compiled XLA extension
jonatanklosko opened this issue · comments
Instead of cuda118 and cuda120 we can just do cuda11 and cuda12, which is what jaxlib does now. Then we just list what are the minimal supported versions for each.