Implementation Details

For now, please see these presentations:

• A. Huebl et al., NREL virtual seminar, December 2022

• A. Huebl et al., “Exascale and ML Models for Accelerator Simulations”, 6th European Advanced Accelerator Concepts workshop (EAAC23), Isola d’Elba, Italy, Sep 17 – 23, 2023 DOI:10.5281/zenodo.8362549

Zero-Copy Data APIs

pyAMReX implements the following standardized data APIs:

• __array_interface__ (CPU)

• __cuda_array_interface__ (CUDA GPU)

• DLPack (coming soon)

These APIs are automatically used when creating “views” (non-copy) numpy arrays, cupy arrays, PyTorch tensors, etc. from AMReX objects such as Array4 and particle arrays.