Li, Jiang c3c2903e72

[Bugfix] Add device assertion to TorchSDPA (#5402)

2024-06-12 12:58:53 -07:00

..

backends

[Hardware] Initial TPU integration (#5292)

2024-06-12 11:53:03 -07:00

ops

[Model] Support MAP-NEO model (#5081)

2024-05-30 19:24:41 -07:00

__init__.py

[Core][2/N] Model runner refactoring part 2. Combine prepare prefill / decode to a single API (#4681)

2024-05-15 14:00:10 +09:00

layer.py

[Bugfix / Core] Prefix Caching Guards (merged with main) (#4846)

2024-05-27 15:18:17 -07:00

selector.py

[Bugfix] Add device assertion to TorchSDPA (#5402)

2024-06-12 12:58:53 -07:00