Paper Notes
首页 标签 列表

JAX

  • JZ-TREE 深度解析:利用 JAX 与 CUDA 协同实现 GPU 友好的邻域搜索与 FoF 聚类

    2026-04-10

    本文深度解析了 JZ-TREE 框架,这是一种基于 Morton 序(Z-Order)平面层级的 GPU 树形结构,通过双树遍历和 JAX/CUDA 协同,在处理超大规模粒子体系时性能优于传统库一个数量级。

    • #GPU Computing
    • #JAX
    • #Nearest Neighbour Search
    • #HPC
    • #Cosmology
    • #Molecular Dynamics
  • gyaradax:深度解析基于 JAX 的可微局部回旋动力学仿真框架

    2026-04-09

    本文深度解析了 gyaradax——一个仅用 3000 行 JAX 代码实现的高性能、可微局部回旋动力学求解器,探讨其在 GPU 加速、自动微分及 AI 辅助开发方面的突破。

    • #回旋动力学
    • #JAX
    • #自动微分
    • #等离子体物理
    • #GPU加速
    • #AI-Agent
  • 深度解析:基于人工神经网络的变分蒙特卡洛方法(ANNVMC)——从理论基础到量子化学应用实践

    2026-03-21

    本文深度解析了 William Freitas 的 ANNVMC 教程,探讨如何利用神经网络作为试探波函数解决多体量子系统基态求解问题,涵盖从一维势场到氢分子的完整演进。

    • #量子化学
    • #变分蒙特卡洛
    • #人工神经网络
    • #深度学习
    • #计算物理
    • #JAX
  • 端到端可微学习:构建适用于 DFT 和 LR-TDDFT 的统一交换相关泛函

    2026-03-07

    本文深度解析了基于 JAX 的 IQC 框架,通过端到端可微工作流优化单一深度学习能量泛函,实现了对基态 DFT 与激发态 LR-TDDFT 的统一描述及分析一致性。

    • #密度泛函理论
    • #自动微分
    • #机器学习泛函
    • #TDDFT
    • #JAX
    • #量子化学
  • 深度解析 WARPAX:利用 JAX 与自动微分攻克曲率驱动时空的观测者鲁棒性能量条件验证

    2026-03-04

    本文深度解析了 WARPAX 工具包,探讨其如何通过 JAX 自动微分和梯度优化,替代传统离散采样法,实现对曲率驱动时空能量条件的观测者鲁棒性验证。

    • #广义相对论
    • #自动微分
    • #JAX
    • #曲率驱动
    • #能量条件
    • #高性能计算
  • WARPAX:曲速驱动时空能量条件观测者鲁棒性深度解析工具

    2026-02-27

    WARPAX是一个基于JAX、GPU加速的开源工具包,用于对曲速驱动时空进行观测者鲁棒的能量条件分析,它通过连续、基于梯度的优化取代了离散采样,并结合Hawking-Ellis代数分类,揭示了单帧分析可能系统性低估能量条件违规的范围和严重程度。

    • #曲速驱动
    • #能量条件
    • #广义相对论
    • #自动微分
    • #JAX

© 2026 Paper Notes.