Framework
Supported versions
Supported OS
Supported Python versions
PyTorch
2.11.0, 2.10.0, 2.9.1
Linux
3.14, 3.13, 3.12, 3.11
2.11.0
Windows
JAX
0.9.1, 0.8.2
Linux
3.14, 3.13, 3.12, 3.11
vLLM
(gfx950, gfx942, gfx1200,
gfx1201, gfx1100,
gfx1101,
gfx1102, gfx1151 GPUs only)
0.19.1
(requires PyTorch 2.10.0)
Linux
3.13