PyTorch example#579

Merged

leofang merged 19 commits intoNVIDIA:mainfrom

msaroufim:pytorch_example

Apr 29, 2025

Commits on Apr 25, 2025

PyTorch example
Mark Saroufim
committed
lint
Mark Saroufim
committed
simplify example
Mark Saroufim
committed
signoff
Mark Saroufim
committed

Commits on Apr 29, 2025

Merge branch 'main' into pytorch_example
msaroufim
authored
test suite changes
Mark Saroufim
committed
Update cuda_core/examples/pytorch_example.py

msaroufim
and
kkraus14
authored
Update cuda_core/examples/pytorch_example.py

msaroufim
and
kkraus14
authored
Update requirements-cu12.txt
msaroufim
authored
Update requirements-cu11.txt
msaroufim
authored
Merge branch 'main' into pytorch_example
msaroufim
authored
Update requirements-cu12.txt

msaroufim
and
kkraus14
authored
Update requirements-cu11.txt

msaroufim
and
kkraus14
authored
Update pytorch_example.py

msaroufim
and
leofang
authored
Update pytorch_example.py

msaroufim
and
leofang
authored
Update cuda_core/examples/pytorch_example.py

msaroufim
and
leofang
authored
remove .item() call
Mark Saroufim
committed
Merge branch 'main' into pytorch_example
Mark Saroufim
authored
Defer CI setup to later
leofang
authored