Publications

Accelerate Coastal Ocean Circulation Model with AI Surrogate
Zelin Xu, Jie Ren, Yupu Zhang, Jose Maria Gonzalez Ondina, Maitane Olabarrieta, Tingsong Xiao, Wenchong He, Zibo Liu, Shigang Chen, Kaleb Smith, and Zhe Jiang
In In 39th IEEE International Parallel & Distributed Processing Symposium. (IPDPS'25 )

Exploring and Evaluating Real-world CXL: Use Cases and System Adoption
Xi Wang, Jie Liu, Jianbo Wu, Shuangyan Yang, Jie Ren, Bhanu Shankar, and Dong Li
In In 39th IEEE International Parallel & Distributed Processing Symposium. (IPDPS'25 )

Machine Learning-Guided Memory Optimization for DLRM Inference on Tiered Memory
Jie Ren, Bin Ma, Benjamin Francis, Ehsan Ardestani, Min Si, and Dong Li
In In 31th IEEE International Symposium on High-Performance Computer Architecture. (HPCA'25 ) (to appear)

ATTNChecker: Highly-Optimized Fault Tolerant Attention for Large Language Model Training
Yuhang Liang, Bo Fang, Xinyi Li, Jie Ren, Ang Li, and Jieyang Chen
In In ACM SIGPLAN Symposium on Principles and Practice of Parallel Programming. (PPoPP'25)

Dissecting Grace Hopper Integrated CPU-GPU System Memory for HPC Applications
Gabin Schieffer, Jennifer Faj, Jie Ren, Jacob Wahlgren, Ivy Peng
In In 53rd International Conference on Parallel Processing. ( ICPP'24 )

Towards Efficient Page Profiling and Migration on Multi-Tiered Large Memory Systems
Jie Ren, Dong Xu, Junhee Ryu, Kwangsik Shin, Daewoo Kim, and Dong Li
In EuroSys XVIII (EuroSys'24)

Enabling Large Dynamic Neural Network Training with Learning-based Memory Management
Jie Ren, Dong Xu, Shuangyan Yang, Jiacheng Zhao, Zhicheng Li, Christian Navasca, Chenxi Wang, Harry Xu, and Dong Li
In 30th IEEE International Symposium on High-Performance Computer Architecture (HPCA'24)

ZeRO-Offload: Democratizing Billion-Scale Model Training
Jie Ren, Samyam Rajbhandari, Reza Yazdani Aminabadi, Olatunji Ruwase, Shuangyan Yang, Minjia Zhang, Dong Li and Yuxiong He
In 2021 USENIX Annual Technical Conference (ATC'21)

Optimizing Large-Scale Plasma Simulations on Persistent Memory-based Heterogeneous Memory with Effective Data Placement Across Memory Hierarchy
Jie Ren, Jiaolin Luo, Ivy Peng, Kai Wu, and Dong Li 
In International Conference on Supercomputing (ICS'21)

Sentinel: Efficient Tensor Migration and Allocation on Heterogeneous Memory Systems for Deep Learning
Jie Ren, Jiaolin Luo, Kai Wu, Minjia Zhang, Hyeran Jeon and Dong Li
In 27th IEEE International Symposium on HighPerformance Computer Architecture(HPCA'21)

Sparta: High-Performance, Element-Wise Sparse Tensor Contraction on Heterogeneous Memory
Jiawen Liu, Jie Ren, Roberto Gioiosa, Dong Li and Jiajia Li
In 26th Principles and Practice of Parallel Programming (PPoPP'21)

ArchTM: Architecture-Aware, High Performance Transaction for Persistent Memory
Kai Wu, Jie Ren, Ivy Peng and Dong Li
In 19th USENIX Conference on File and Storage Technologies (FAST'21)

HM-ANN: Efficient Billion-Point Nearest Neighbor Search on Heterogeneous Memory
Jie Ren, Minjia Zhang and Dong Li
In 34th Conference on Neural Information Processing Systems(NeurIPS'20)

Exploring Non-Volatility of Non-Volatile Memory for High Performance Computing Under Failures
Jie Ren, Kai Wu and Dong Li
In IEEE International Conference on Cluster Computing (Cluster'20)

Ribbon: High Performance Cache Line Flushing for Persistent Memory
Kai Wu, Ivy B. Peng, Jie Ren and Dong Li
In 29th International Conference on Parallel Architectures and Compilation Techniques (PACT'20)

Sparta: High-Performance, Element-Wise Sparse Tensor Contraction on Heterogeneous Memory
Jiawen Liu, Jie Ren, Roberto Gioiosa, Dong Li and Jiajia Li
In 26th Principles and Practice of Parallel Programming (PPoPP'21)

Demystifying the Performance of HPC Scientific Applications on NVM-based Memory Systems
 Ivy Peng, Kai Wu, Jie Ren, Dong Li and Maya Gokhale.
In 34th IEEE International Parallel and Distributed Processing Symposium (IPDPS'20)

Runtime Data Management on Non-Volatile Memory-based Heterogeneous Memory for Task-Parallel Programs
 Kai Wu, Jie Ren, and Dong Li
In 30th ACM/IEEE International Conference for High Performance Computing, Networking, Storage and Analysis (SC'18)

Opera: Data Access Pattern Similarity Analysis To Optimize OpenMP Task Affinity
Jie Ren, Chunhua Liao, and Dong Li
In 24th International Workshop On High-level Parallel Programming Models And Supportive Environments (HIPS'18),

Understanding Application Recomputability without Crash Consistency in Non-Volatile Memory
Jie Ren, Kai Wu, and Dong Li
In Workshop on Memory Centric Programming for HPC (MCHPC'18)