AI Models for Earthsystem
Role details
Job location
Tech stack
Job description
- Contributes technical experience through analysis and support for programs and projects associated with machine learning, HPC, and computational problems related to earth system science and other dynamical systems.
- Develops and evaluates machine learning/computational approaches, synthesis activities, computational tools, compiling results, contributes to reports, publications, and documentation.
- In particular, this position will assist on projects related to applying and developing machine learning-based weather models for the S2S time frame with an emphasis on generative AI techniques, evaluating such models, and working with a team of scientists.
Requirements
The ideal candidate has a PhD in geophysical sciences, computer science, or machine learning with experience in developing and verifying deep learning-based models for large dynamical systems (e.g. weather). Some familiarity in data and model parallelisms for distributed training on large GPU-based machines is essential. who have experience with diffusion-based or other generative AI methods and multi-modal embeddings, as well as a background in atmospheric science, especially weather modeling., * Experience with deep learning, PyTorch/ JAX, and scaling deep learning models to large GPU-based machines.
- Experience building, training and running inferences with large AI foundation models for science domain.
- Technical knowledge in using HPC systems for visualization and analysis.
- Knowledge of large, dynamical systems (preferably the atmosphere), is desirable.
- Skills in clear, concise writing of technical papers, and interacting and communicating effectively with colleagues.
- Some problem-solving skills.
- Organizational skills and flexibility in coordinating a broad spectrum of activities.
- Knowledge of atmospheric dynamics, process scale models, and numerical computation techniques is preferred.
- Experience in scientific programming and data analysis.
- Knowledge of using atmospheric observational datasets, data assimilation techniques, and statistics is preferred.
- Familiarity sub-seasonal-to-seasonal modeling and or coupled atmosphere-ocean modeling is desirable.
- Ability to work and communicate with stakeholders from public and private sectors.
- Ability to model Argonne's core values of impact, safety, respect, integrity, and teamwork.
Minimum Education/Experience Requirements: PhD Degree or their equivalents in geophysical sciences, computer science, machine learning, or a related field.