File size: 280 Bytes
6f4b39f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | [project]
name = "my-softmax-function"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.14"
dependencies = [
"nvidia-cutlass-dsl>=4.6.1",
"torch>=2.13.0",
]
[dependency-groups]
dev = [
"kernels",
"pytest",
]
|