numpy
torch
triton

[dev]
pytest
black
isort
flake8
entmax
