Motif-Technologies
/

optimizer

Model card Files Files and versions

optimizer / test

46.5 kB

15 contributors

History: 17 commits

wyldecat's picture

Apply pre-commit formatting (isort) [skip-build]

96b287c 5 days ago

README.md

1.3 kB

Support param group with various placements (#13) 4 months ago
__init__.py

0 Bytes

Support param group with various placements (#13) 4 months ago
conftest.py

5.58 kB

Add torch.compile, CUDA graph, and compiled momentum [skip-build] 6 days ago
optimizer

23 Bytes

Support param group with various placements (#13) 4 months ago
pytest.ini

45 Bytes

Support param group with various placements (#13) 4 months ago
run_test.sh

74 Bytes

Support param group with various placements (#13) 4 months ago
run_test_moe.sh

124 Bytes

Refactor pipeline to async generator pattern (#16) 11 days ago
test_muon.py

13 kB

Add uneven shard correctness test [skip-build] 6 days ago
test_muon_moe.py

13.1 kB

Apply pre-commit formatting (isort) [skip-build] 5 days ago
test_normalize_fqn.py

3.9 kB

Update tests for MoE and parallel optimizations [skip-build] 6 days ago
utils.py

9.36 kB

Update tests for MoE and parallel optimizations [skip-build] 6 days ago