Woosuk Kwon 91b47e3f2f JAX-based TPU worker

2024-04-16 17:37:11 +00:00

..

__init__.py

Change the name to vLLM (#150)

2023-06-17 03:07:40 -07:00

cache_engine.py

Merge branch 'main' into woosuk-tpu

2024-04-10 07:51:35 +00:00

cpu_model_runner.py

[Hardware][Intel] Isolate CPUModelRunner and ModelRunner for better maintenance (#3824)

2024-04-11 11:56:49 -07:00

cpu_worker.py

[Bugfix] More type hint fixes for py 3.8 (#4039)

2024-04-12 21:07:04 -07:00

model_runner.py

[Frontend] [Core] feat: Add model loading using tensorizer (#3476)

2024-04-13 17:13:01 -07:00

neuron_model_runner.py

[Misc] Minor fix in KVCache type (#3652)

2024-03-26 23:14:06 -07:00

neuron_worker.py

[Bugfix] More type hint fixes for py 3.8 (#4039)

2024-04-12 21:07:04 -07:00

tpu_worker.py

JAX-based TPU worker

2024-04-16 17:37:11 +00:00

worker_base.py

[Bugfix] More type hint fixes for py 3.8 (#4039)

2024-04-12 21:07:04 -07:00

worker.py

[Frontend] [Core] feat: Add model loading using tensorizer (#3476)

2024-04-13 17:13:01 -07:00