Integrating physical units into high-performance AI-driven scientific computing

Chaoming Wang, Sichao He, Shouwei Luo, Yuxiang Huan and Si Wu ()
Additional contact information
Chaoming Wang: Peking University
Sichao He: Peking University
Shouwei Luo: Shanghai Jiao Tong University
Yuxiang Huan: Guangdong Institute of Intelligence Science and Technology
Si Wu: Peking University

Nature Communications, 2025, vol. 16, issue 1, 1-14

Abstract: Abstract Artificial intelligence is revolutionizing scientific research across various disciplines. The foundation of scientific research lies in rigorous scientific computing based on standardized physical units. However, current mainstream high-performance numerical computing libraries for artificial intelligence generally lack native support for physical units, significantly impeding the integration of artificial intelligence methodologies into scientific research. To fill this gap, we introduce SAIUnit, a system designed to seamlessly integrate physical units into scientific artificial intelligence libraries, with a focus on compatibility with JAX. SAIUnit offers a comprehensive library of over 2000 physical units and 500 unit-aware mathematical functions. It is fully compatible with JAX transformations, allowing for automatic differentiation, just-in-time compilation, vectorization, and parallelization while maintaining unit consistency. We demonstrate SAIUnit’s applicability and effectiveness across diverse artificial intelligence-driven scientific computing domains, including numerical integration, brain modeling, and physics-informed neural networks. Our results show that by confining unit checking to the compilation phase, SAIUnit enhances the accuracy, reliability, interpretability, and collaborative potential of scientific computations without compromising runtime performance. By bridging the gap between abstract computing frameworks and physical units, SAIUnit represents a crucial step towards more robust and physically grounded artificial intelligence-driven scientific computing.

Date: 2025
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.nature.com/articles/s41467-025-58626-4 Abstract (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:nat:natcom:v:16:y:2025:i:1:d:10.1038_s41467-025-58626-4

Ordering information: This journal article can be ordered from
https://www.nature.com/ncomms/

DOI: 10.1038/s41467-025-58626-4

Access Statistics for this article

Nature Communications is currently edited by Nathalie Le Bot, Enda Bergin and Fiona Gillespie

More articles in Nature Communications from Nature
Bibliographic data for series maintained by Sonal Shukla () and Springer Nature Abstracting and Indexing ().