NWP AI/ML Workflow Engineer

Axiom Consultants; College Park (MD); On-site; USD 110,000 - 170,000; Full time

Benefits offered by this job

Medical insurance

Dental insurance

Life insurance

401(k) matching

Job summary

Axiom is seeking candidates to support NOAA EMC in College Park, MD, working on the Seasonal Forecast System (SFS). The role focuses on designing and maintaining AI/ML-enabled NWP workflows that run on HPC and Cloud platforms, in collaboration with software engineers and scientists in an Agile setting.

The successful candidate will develop new capabilities, modernize infrastructure, and contribute to end-to-end workflow optimization for forecast guidance generation.

Qualifications

• Background in Physical Sciences, Computer Sciences, or Mathematics with proven experience in numerical modeling.
• Strong knowledge of AI/ML and available packages and tools.
• Strong knowledge of parallel computing, especially using GPUs. Experience in developing and optimizing code for HPC systems. Experience with designing and deploying applications on Cloud platforms
• Knowledge of Object Oriented Programming concepts with emphasis on the development of efficient and scalable code. Familiarity with modern Fortran, C++, MPI, and scientific computing libraries is a plus
• Strong analytical and problem-solving skills, with the ability to troubleshoot complex software systems
• Experience with Python and Unix Shell Scripting
• Experience with version control systems, such as Git and GitHub
• Excellent verbal and written communication skills, with the ability to convey technical concepts to both technical and non-technical stakeholders

Responsibilities

• Develop innovative AI/ML-based methods and tools to improve and optimize the workflow.
• Design, develop, and maintain workflow features to enable seamless deployment of NWP applications to HPC and Cloud-based platforms
• Ensure the workflow efficiently utilizes both HPC resources and Cloud platforms
• Work closely with cross-functional teams in an Agile-style development environment to achieve project milestones and deliver high-quality software
• Follow software development best practices
• Create clear and comprehensive documentation for software and processes

Skills

AI/ML

HPC

Python

Unix Shell

Git/GitHub

Fortran

C++

MPI

GPU Programming

Education

Physical Sciences/CS/Math background

Tools

CUDA

Cloud platforms

GPU toolchains

Job description

Axiom is seeking candidates to support NOAA EMC in College Park, MD, working on the Seasonal Forecast System (SFS). The role focuses on designing and maintaining AI/ML-enabled NWP workflows that run on HPC and Cloud platforms, in collaboration with software engineers and scientists in an Agile setting.

The successful candidate will develop new capabilities, modernize infrastructure, and contribute to end-to-end workflow optimization for forecast guidance generation.