Senior Research Software Engineer
Core
Design, build, and maintain ML/AI systems and research software to support faculty-led research projects in domains like climate science and global health.
Role type
Senior Research Software Engineer (HPC/ML Infrastructure)
Builds
Production-grade research software, optimized ML training/inference pipelines, and reusable technical patterns for AWS and HPC environments.
Domain
Academic research, High-Performance Computing, Artificial Intelligence/Machine Learning
Deliverable
production ML models
Required skills
Python, C/C++, CUDA, OpenMP, OpenCL, MLIR, LLVM, XLA, IREE, TVM, Halide, cuBLAS, cuDNN, CUTLASS, HIP, ROCm, PyTorch, TensorFlow, JAX, distributed computing (DP/TP/SP/CP/EP), profiling, performance optimization, Git, AWS, HPC
Preferred skills
Compiler infrastructure contribution, secure/privacy-constrained data engineering, interdisciplinary research experience
Technologies
AWS, HPC, CUDA, OpenMP, OpenCL, MLIR, LLVM, XLA, IREE, TVM, Halide, PyTorch, TensorFlow, JAX, ONNX, TensorRT
Responsibilities
Develop and optimize machine learning training and inference pipelines for accelerator-based systems; Apply systems- and compiler-level optimizations (loop transformations, vectorization, parallelization); Implement and optimize kernels using CUDA or accelerator-specific models; Collaborate with researchers and AWS partners on systems engineering and performance tuning; Support execution in AWS cloud and HPC environments including large-scale model training and debugging; Contribute to compiler and IR frameworks.
Seniority
Senior, hands-on IC
