Antoni Baum 3b682179dd

[Core] Add AttentionState abstraction (#7663)

2024-08-20 18:50:45 +00:00

..

__init__.py

[Speculative decoding 2/9] Multi-step worker for draft model (#2424)

2024-01-21 16:31:47 -08:00

test_encoder_decoder_model_runner.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -07:00

test_model_input.py

[Core] Add AttentionState abstraction (#7663)

2024-08-20 18:50:45 +00:00

test_model_runner.py

[Core] Optimize SPMD architecture with delta + serialization optimization (#7109)

2024-08-18 17:57:20 -07:00

test_swap.py

[Core] Pipeline Parallel Support (#4412)

2024-07-02 10:58:08 -07:00