DeepMind 宣布与 Commonwealth Fusion Systems 建立研究合作，旨在加速清洁、安全的聚变能源实现。聚变是太阳的能量来源过程，能提供清洁、丰富的能源，且不产生长寿命放射性废物。在地球上实现聚变需要将等离子体稳定在超过 1 亿摄氏度的温度下，这是一个高度复杂的物理问题，DeepMind 正利用人工智能来解决。
CFS 是聚变能源领域的全球领导者，其紧凑而强大的托卡马克机器 SPARC 旨在成为历史上第一个产生净聚变能量的磁聚变机器，即实现“盈亏平衡”里程碑。DeepMind 与 CFS 的合作基于其先前使用 AI 成功控制等离子体的突破性工作，包括与瑞士等离子体中心的合作，展示了深度强化学习可以控制托卡马克的磁体以稳定复杂等离子体形状。
为了覆盖更广泛的物理范围，DeepMind 开发了 TORAX，这是一个用 JAX 编写的快速且可微分的等离子体模拟器。TORAX 将帮助 CFS 团队在 SPARC 启动前通过运行数百万次虚拟实验来测试和优化操作计划，并能在首批数据到达后快速调整计划。该软件已成为 CFS 日常工作的关键部分，帮助他们理解等离子体在不同条件下的行为，节省时间和资源。
操作托卡马克涉及调整各种“旋钮”，如磁线圈电流、燃料注入和加热功率。手动寻找最优设置以产生最多能量，同时保持在操作限制内，可能效率低下。DeepMind 的 AI 代理结合 TORAX 和强化学习或进化搜索方法，可以在模拟中探索大量潜在操作场景，以优化性能。