2025年12月7日
JAX核心设计解析：函数式编程让代码更可控
摘要： 很多人刚接触JAX都会有点懵——参数为啥要单独传？随机数还要自己管key？这跟PyTorch的画风完全不一样啊。 其实根本原因就一个：JAX是函数式编程而不是面向对象那套，想明白这点很多设计就都说得通了。 先说个核心区别 PyTorch里，模型是个对象，权重藏在里面，训练的时候自己更新自己。这是典型 阅读全文
posted @ 2025-12-07 22:01 deephub 阅读(5) 评论(0) 推荐(0)