|
inference
|
850b72d3ea
make StatefulShardedModel callable, add some tests for mlx sharded inference
|
1 rok temu |
|
networking
|
445eda156c
dynamically assign shards to nodes deterministically weighted by memory
|
1 rok temu |
|
orchestration
|
445eda156c
dynamically assign shards to nodes deterministically weighted by memory
|
1 rok temu |
|
topology
|
36b8456798
collect global topology with local peer visibility, ring memory weighted partitioning strategy
|
1 rok temu |
|
.gitignore
|
850b72d3ea
make StatefulShardedModel callable, add some tests for mlx sharded inference
|
1 rok temu |
|
example_user.py
|
36b8456798
collect global topology with local peer visibility, ring memory weighted partitioning strategy
|
1 rok temu |
|
example_user_2.py
|
445eda156c
dynamically assign shards to nodes deterministically weighted by memory
|
1 rok temu |
|
main.py
|
36b8456798
collect global topology with local peer visibility, ring memory weighted partitioning strategy
|
1 rok temu |
|
main_dynamic.py
|
445eda156c
dynamically assign shards to nodes deterministically weighted by memory
|
1 rok temu |
|
requirements.txt
|
3a66a0a4a8
add requirements.txt
|
1 rok temu |
Alex Cheema
850b72d3ea
make StatefulShardedModel callable, add some tests for mlx sharded inference