AI Infrastructure Engineer

Job Summary
:
Spectral AI is focused on developing AI models for medical imaging, and they are seeking an AI Infrastructure Engineer to design, build, and maintain the infrastructure that supports these models. The role involves working with multi-modal data pipelines and collaborating with various teams to ensure the deployment of validated AI systems.

Responsibilities
:

• Architect and maintain scalable infrastructure for medical imaging AI workflows, including MSI/RGB data ingestion, preprocessing, training, evaluation, and deployment

• Build reliable model serving pipelines that integrate with clinical software systems

• Improve image chain & algorithm performance compared to initial benchmarks

• Optimize CUDA kernels for maximum GPU utilization and performance

• Support acceleration strategies for workflow-constrained clinical use cases

• Work closely with Data Scientists, Computer Vision Engineers, and Software Engineers to operationalize AI models

• Partner with clinical, QA, and regulatory teams to ensure infrastructure supports validated and auditable AI workflows

Qualifications
:
Required
:

• Master's or PhD in Computer Science, Engineering, or related field

• Minimum of 1 year of industry experience supporting production-level AI/ML systems, infrastructure, or deployment pipelines

• Proficient in Python and C++

• Experience with cloud platforms (AWS, GCP, or Azure)

• Experience with model serving and inference frameworks (Tensor

RT, ONNX Runtime)

• Hands-on experience with containerization (Docker, Kubernetes) and CI/CD pipelines

• Familiarity with ML frameworks like PyTorch or Tensor Flow from a systems perspective

• Knowledge of CUDA programming and GPU optimization

Preferred
:

• Deep understanding of neural network architectures and training methodologies

• Experience in productizing medical image algorithms on GPU platforms

• Publications in top-tier conferences (CVPR, ICCV, ECCV, NeurIPS, ICML)

Company
:
Spectral AI specializes in predictive analytics with a platform that uses proprietary artificial intelligence (AI). Founded in 2009, the company is headquartered in Dallas, USA, with a team of 51-200 employees. The company is currently Growth Stage.