Chengji Yao 8e882ffdc0

[Bugfix][TPU] fix moe custom kernel import (#18853) ...

Signed-off-by: Chengji Yao <chengjiyao@google.com>

2025-05-28 19:34:19 -07:00

..

guided_decoding

Enable Pydantic mypy checks and convert configs to Pydantic dataclasses (#17599)

2025-05-28 12:46:04 +00:00

layers

[Bugfix][TPU] fix moe custom kernel import (#18853)

2025-05-28 19:34:19 -07:00

model_loader

[Doc] Move examples and further reorganize user guide (#18666)

2025-05-26 07:38:04 -07:00

models

Prevent the cross-encoder logic from being applied to classification tasks (#18838)

2025-05-28 19:16:17 -07:00

__init__.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

custom_op.py

Update some more deprecated type hinting (#17998)

2025-05-12 23:49:33 +00:00

parameter.py

[Kernel] Support Microsoft Runtime Kernel Lib for our Low Precision Computation - BitBLAS (#6036)

2025-04-22 09:01:36 +01:00

pooling_metadata.py

Update some more deprecated type hinting (#17998)

2025-05-12 23:49:33 +00:00

sampling_metadata.py

Update some more deprecated type hinting (#17998)

2025-05-12 23:49:33 +00:00

utils.py

Update some more deprecated type hinting (#17998)

2025-05-12 23:49:33 +00:00