T
traeai
Sign in

产品

Pallas

TPU专用的自定义内核框架

已跟踪 3 条高相关材料

TraeAI 观察

相关材料

已收录 3 条与 Pallas 相关的内容,按评分排序。

Google Developers Blog 图标

Announcing New vLLM and SGLang Inference Engines on TPU

Google Developers Blog1557 字 (约 7 分钟)
85

Google推出基于TPU的vLLM和SGLang推理引擎,通过TorchTPU实现PyTorch原生支持,优化大模型服务性能。

入选理由:TorchTPU使TPU成为原生PyTorch设备,支持自定义Pallas内核

FeaturedArticle#TPU#PyTorch#推理引擎#AI#大模型英文
Helion on TPU: Towards Hardware Heterogeneous Kernel Authoring

Helion on TPU: Towards Hardware Heterogeneous Kernel Authoring

PyTorch Blog2268 字 (约 10 分钟)
85

Helion通过高级DSL和自动调优,使TPU内核开发更高效,性能达838 TFLOPs。

入选理由:Helion生成的TPU内核在flash attention任务中达到838 TFLOPs,接近79%的MFU。

FeaturedArticle#PyTorch#TPU#DSL#性能优化英文
[AINews] How to land a job at a frontier lab (on Pretraining)

How to Land a Job at a Frontier Lab (on Pretraining)

Latent Space1926 字 (约 8 分钟)
85

Vlad Feinberg's guide highlights mastering LLM kernel-level tuning and MoE architecture optimization as critical for entering frontier labs, while agent automation and observability emerge as infrastructure trends.

入选理由:掌握LLM内核调优(如JAX/Pallas)是进入前沿实验室的最直接路径,需能手写代码实现MoE层优化

FeaturedArticle#LLM Kernel Tuning#MoE Architecture#Agent Automation#DeepMind#LangChain英文

跨材料问答 · Pallas

回答基于:Pallas 相关 3 条材料
    0 / 500

    AI may generate inaccurate information. Please verify important content.