ChatCrawlersearch across public Telegram Open the app
P

Python 中文交流群

snapshot for January 2026
‹ January 2026 ›×
26 January 2026
J
哎算了，tensorflow 这东西谁见了都头痛
S
（ 一直没去学jax
不过差距仅限于深层小模型。层次不深，或者每一层计算量很大的话差距就不明显了。
A
根本原因在PyTorch.compile的loop是还是跑在Python上面的，只是编译成了C++的C Extension。层次一深cuda->c++->python->c++->cuda的路径上面python段会产生很多开销。 jax.jit编译之后整个函数就都完全不在Python里面跑了，层次深浅对整体影响不大。 (23年的测试了，不知道现在PyTorch有没有优化过了）
🅼
🅼🆁🆂🆆🅸🆃🅲🅷
aaaa llll-ある在饿死的边缘徘徊根本原因在PyTorch.compile的loop是还是跑在Python上面的，只是编译成了C++的C Extension。层次一深cuda->c++->python->c++->cuda的路径上面python段会产生很多开销。 jax.jit编译之后整个函数就都完全不在Python里面跑了，层次深浅对整体影响不大。 (23年的测试了，不知道现在PyTorch有没有优化过了）
你这研究挺深啊，一般都是定义一下网络结构就开始摸鱼炼丹了反正都要等很久
S
S.
aaaa llll-ある在饿死的边缘徘徊根本原因在PyTorch.compile的loop是还是跑在Python上面的，只是编译成了C++的C Extension。层次一深cuda->c++->python->c++->cuda的路径上面python段会产生很多开销。 jax.jit编译之后整个函数就都完全不在Python里面跑了，层次深浅对整体影响不大。 (23年的测试了，不知道现在PyTorch有没有优化过了）
听起来没有很难解决。之前这个讨论dl群好像也进行过，过程我不记得了结论好像是效率差不多了
Open in Telegram Каталог площадок Искать в ChatCrawler

A snapshot of an open public feed from the search index ChatCrawler — “Google for public Telegram”; refreshed as the venue is crawled. Times are UTC.

Public content only, official Telegram API. About · FAQ · What we do not do · Remove a page · Catalog