3 people 9112b443a0 [Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011) ...

Signed-off-by: Siyuan Liu <lsiyuan@google.com>
Co-authored-by: Hossein Sarshar <hossein.sarshar@gmail.com>
Co-authored-by: Chengji Yao <chengjiyao@google.com>

2025-06-03 00:06:20 +00:00

..

__init__.py

[V1][Spec Decoding] Use model_loader.get_model() to load models (#18273)

2025-05-23 02:05:44 +00:00

base_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

bitsandbytes_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

default_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

dummy_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

gguf_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

neuron.py

Update deprecated type hinting in model_loader (#18130)

2025-05-15 04:00:21 -07:00

neuronx_distributed.py

[Neuron] Add Multi-Modal model support for Neuron (#18921)

2025-05-31 10:39:11 +00:00

runai_streamer_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

sharded_state_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

tensorizer_loader.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

tensorizer.py

[Core] Support inplace model weights loading (#18745)

2025-06-02 17:38:50 +08:00

tpu.py

[Hardware][TPU] Initial support of model parallelism with single worker using SPMD (#18011)

2025-06-03 00:06:20 +00:00

utils.py

[Bugfix] Fix transformers model impl ignored for mixtral quant (#18602)

2025-05-23 05:54:13 -07:00

weight_utils.py

Fix an error in dummy weight loading for quantization models (#18855)

2025-05-29 03:07:20 -07:00