High-level GPU code: a case study examining JAX and OpenMP. - Demeure, Nestor; Kisner, Theodore; Keskitalo, Reijo; Thomas, Rollin; Borrill, Julian; Bhimji, Wahid (Proceedings of the SC '23 Workshops of The International Conference on High Performance Computing, Network, Storage, and Analysis, 2023)