numpy
torch
einops
ninja
packaging
opt-einsum
pykeops

[conv1d]
causal-conv1d

[dev]
pytest
black
isort
mypy
