About this role
Supports development and optimization of AI/ML workflows and neural network models for climate and Earth system science, and deploys and optimizes large-scale training and inference on HPC and GPU-accelerated systems. Provides user engagement, training, documentation, grant support, and consulting to researchers to advance computational research.
Skills for this role
PythonC++PyTorchTensorFlowNeural network architecturesDeep learningLinux/UNIXHigh-performance computing (HPC)SLURMGPU-accelerated systemsDistributed trainingModel scalingPerformance optimizationNVIDIA NsightNumPySciPypandasxarrayscikit-learnNetCDFHDF5WRFCESMGitDockerMPIOpenMPVision Transformers (ViTs)Generative modelingFlow matching''Stochastic interpolants''Performance analysis''Compiler optimz