AI Computational Scientist – Senior Software Engineer

The University of Texas at Austin•Austin, UT

1d•Onsite

About The Position

The AI Computational Scientist will provide technical expertise and coordination to advance the use of artificial intelligence and machine learning in Earth and planetary science research across the Jackson School of Geosciences and its three units: the Department of Earth and Planetary Sciences, the Institute for Geophysics, and the Bureau of Economic Geology. The position is housed in the Office of the Dean and reports to the Special Advisor to the Dean for AI and Computing. It complements and supports the existing computational scientists in the units.

Requirements

Master's degree in Earth sciences, computational sciences, applied mathematics, computer science, engineering, physics, or a related field.
At least three years of experience in software development on complex systems.
Production experience with at least two programming languages, including scientific Python.
Demonstrated experience developing, training, and evaluating machine learning models using frameworks such as PyTorch, JAX, TensorFlow, or similar technologies.
Experience with Linux-based scientific computing environments and high-performance computing systems.
Experience using version control systems and collaborative software development practices.
Demonstrated ability to work independently, manage multiple priorities, and meet established deadlines.
Excellent oral, written, and interpersonal communication skills, with the ability to collaborate effectively with multidisciplinary research teams.
Relevant education and experience may be substituted as appropriate.

Nice To Haves

Ph.D. in Earth sciences, computational sciences, applied mathematics, computer science, engineering, physics, or a related field.
Demonstrated experience with scientific machine learning methods, including neural surrogates or emulators, physics-informed neural networks, operator learning, or related approaches.
Research experience in Earth and planetary sciences or significant experience working with geoscience datasets.
Experience developing computational training materials, instructional modules, workshops, or course content.
Experience integrating and managing heterogeneous scientific datasets and data formats.
Familiarity with cloud computing platforms and advanced computing resources, including the TACC ecosystem and related research computing infrastructure.
Record of peer-reviewed scholarly publications and/or contributions to open-source software projects.
Proficiency in additional programming languages beyond Python.

Responsibilities

Partner with faculty, research scientists, postdoctoral scholars, graduate students, and unit computational scientists across the Jackson School of Geosciences to accelerate research through the application of artificial intelligence and machine learning methods.
Design, implement, and evaluate machine learning, scientific AI, and data integration workflows for Earth and planetary science research applications.
Develop and apply scientific AI approaches, including neural surrogates and emulators, graph-network-based simulators, physics-informed neural networks, and foundation-model-based tools for geoscience data.
Collaborate directly with research groups through embedded engagements to enhance computational workflows, transfer technical expertise, and support adoption of advanced AI methods.
Develop, review, test, and debug scientific software and computational workflows in partnership with researchers.
Collaborate with unit computational staff to provide guidance on AI-assisted software development, including evaluation, verification, effective use, and limitations of code-generation tools.
Curate and maintain shared resources, including inventories of AI-ready geoscience datasets, benchmark problems, and reference implementations for common scientific AI applications.
Collaborate with faculty to develop computational learning materials, including notebooks, datasets, exercises, and instructional modules that introduce artificial intelligence and machine learning concepts in undergraduate and graduate geoscience courses.
Support workshops, training activities, and educational initiatives focused on AI methods, scientific computing, and AI-assisted programming.
Provide technical expertise and consultation to promote responsible and effective adoption of AI technologies in educational settings.
Collaborate with the Dean's Office, academic units, and university partners to identify and advance AI-related opportunities across the Jackson School of Geosciences.
Represent the school in university-level technical AI working groups and communicate opportunities related to computing resources, software licensing, seed funding, and related initiatives.
Work with Dean’s staff to evaluate, develop, and help implement administrative AI approaches to facilitate workflows and coordinate technology transfer and training for JSG staff.
Track, document, and report on AI-related activities, needs, and emerging opportunities across the school.
Perform other related duties as assigned.
This position does not administer shared computing infrastructure, provide general information technology support, or manage generative AI model training programs.