Combining jax-triton on Windows (triton-windows) ?

[DiagRisker]

Hello dear contributors,

I am very interested in combining jax and triton, however I have the library triton-windows installed on my side.
When trying to pip install jax-triton, the installation fails because it is looking for the triton package (per say not with -windows suffix)

> pip install 'jax-triton @ git+https://github.com/jax-ml/jax-triton.git'
[...]
ERROR: Could not find a version that satisfies the requirement triton>=3.3 (from jax-triton) (from versions: none)
ERROR: No matching distribution found for triton>=3.3
> pip install --upgrade triton-windows
Requirement already satisfied: triton-windows in [...]\python313\lib\site-packages (3.4.0.post20)
Requirement already satisfied: setuptools>=40.8.0 in [...]\lib\site-packages (from triton-windows) (80.9.0)

Any idea to make it work? (should I contact instead triton-windows admins https://github.com/woct0rdho/triton-windows to see if they can make it compatible just from lib internal naming?)
perhaps recent contributors ? @superbobry

Thanks in advance!

[DiagRisker]

I see few contributors on the repository,
I hope a gentle ping on the topic is ok for @chr1sj0nes , @chsigg , @superbobry , @Moerafaat

[superbobry]

We can probably add triton-windows as a platform-specific dependency, but it is a bit of a stretch given that Triton officially does not support Windows.

If the issue is just being able to pip install jax-triton, you could force pip to do that by passing --no-deps.

[DiagRisker]

Thank you, it works! (my rookie mystake)

Cheers,