Motif-Technologies
/

optimizer

Model card Files Files and versions

optimizer / build

66.8 MB

9 contributors

History: 2 commits

iamwyldecat's picture

fix(muon): delete intermediate tensors immediately to lower peak mem usage

bdd2678 3 months ago

torch26-cxx11-cu118-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx11-cu124-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx11-cu126-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx11-rocm62-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx98-cu118-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx98-cu124-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch26-cxx98-cu126-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch27-cxx11-cu118-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch27-cxx11-cu126-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch27-cxx11-cu128-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago
torch27-cxx11-rocm63-x86_64-linux
fix(muon): delete intermediate tensors immediately to lower peak mem usage 3 months ago