在机器学习中，形状错误是一种难以调试的问题，通常只有在运行模型时才能发现。通过将形状检查作为类型，您可以消除大部分这种挑战，并更快地开发。Jaxtyping 是一个库，它使用 JAX 的类型系统来提供形状检查，从而...

谷歌利用MaxText和Jax训练技术，同时在超过5万个芯片上进行了LLM（语言模型）训练任务，保持了66%以上的利用率，成为了最佳MFU的强有力竞争者。该项目主要基于专有技术，但其中展示了一些有趣的细节，对于其他大规模...

Waymo近日推出了全新的自动驾驶仿真器 Waymax 以评估代理在自动驾驶任务中的表现。该仿真器完全采用 Jax 编写。

SDXL是一种最先进的图像生成模型，虽然强大，但运行速度较慢。然而，Google与HuggingFace最近在Google的研究语言Jax中展示了它在新型TPU芯片上的运行速度。通过这种新的硬件加速，SDXL的性能得到了极大的提升，可以...

DeepMind最近推出了一款名为SynJax的新工具，该工具被设计为处理结构概率分布，如树、序列、分割、对齐等对象上的概率。SynJax的出现，无疑为我们在处理这类问题上提供了更多的便利。该工具不仅包含了许多实用的工具...

近日，苹果公司发布了一款新的深度学习库AX，该库基于Jax构建。值得一提的是，AX支持基于编译器的并行性，可实现大规模训练，这是许多其他流行框架所不具备的功能。在过去的几年中，深度学习领域的研究和应用取得了...

苹果正在创建名为Ajax的AI框架以及一种名为Apple GPT的聊天机器人服务，以求在大型语言模型领域中竞争优势。虽然苹果的AI努力仍然处于实验阶段，但业内人士预计明年会有重大的公告。Apple GPT是苹果全新的聊天机器人...

从头开始训练基础模型是一项具有挑战性的任务。斯坦福大学的语言模型团队发布了一个在Jax上的框架，使这个过程更加简化。它提供了多种数据和模型并行策略，简化了编程过程。