Re:Run Llama2-70B in Web Browser with WebGPU Acceleration
by crowwork on Monday July 24, 2023 @02:36PM
(#36852431)
Checkout the latest docs https://mlc.ai/mlc-llm/docs/ MLC started with demos and it evolved lately, with API integrations, documentations into an inference solution that everyone can reuse for universal deployments
Re:Run Llama2-70B in Web Browser with WebGPU Acceleration
by brucethemoose2 on Monday July 24, 2023 @02:43PM
(#36852516)
Its been awhile since I looked into this, thanks.
As a random aside, I hope y'all publish a SDXL repo for local (non webgpu) inference. SDXL is too compute heavy to split/offload to cpu like Llama.cpp, but less ram heavy than llms, and I'm thinking it would benefit from TVM's "easy" quantization.
It would be a great backend to hook into the various web UIs, maybe with the secondary model loaded on an IGP.
Re:Run Llama2-70B in Web Browser with WebGPU Acceleration
by junrushao1994 on Monday July 24, 2023 @02:15PM
(#36852111)
I don't think TVM advertised a lot on its full capabilities, for example, high-perf codegen for dynamic shapes without auto-tuning, or auto-tuning-based codegen, at least in the past few years, and that might be one of the factors it doesn't got a lot of visibility.
Re:Run Llama2-70B in Web Browser with WebGPU Acceleration
by brucethemoose2 on Monday July 24, 2023 @02:20PM
(#36852185)
I think this is true of AI compilation in general. Torch MLIR, AITemplate and really everything here fly under the radar:
https://github.com/merrymercy/awesome-tensor-compilers#open-...
But... It isn't used much outside MLC? And MLC's implementations are basically demos.
I dunno why. AI inference communities are dying for fast multiplatform backends without the fuss of PyTorch.