Edge AI/ML Engineer — Real-Time Wireless Inference

Booz Allen Hamilton; Corridor North (MD); On-site; USD 99,000 - 225,000; Full time

Job summary

Booz Allen Hamilton seeks an AI/ML engineer to join a team building a converged platform that runs real-time wireless signal processing and AI inference on GPU hardware. You will design, train, and deploy models, build data pipelines, and containerize services on Kubernetes.

The role requires 3+ years with PyTorch/TensorFlow, Python, GPU computing, and the ability to obtain a Secret clearance; a STEM degree with extensive AI/ML experience is required.

Qualifications

• Experience training and evaluating ML models with PyTorch or TensorFlow.
• 3+ years of Python software development, including testing and VCS.
• Experience deploying ML models as services in containerized environments.
• Ability to obtain a Secret clearance.
• Strong collaboration in Agile, cross-functional teams.

Responsibilities

• Design, train, and evaluate ML models for wireless network use cases.
• Build data pipelines ingesting telemetry into training datasets.
• Optimize models for real-time GPU inference and benchmarking.
• Package models as containerized services on Kubernetes with NVIDIA tools.
• Collaborate with engineers to translate mission requirements into deployable models.

Skills

ML model development

Python development

GPU computing

Docker / Kubernetes

Agile / cross-functional team

Technical documentation

Education

Bachelor's degree in STEM

Master's degree in STEM

8+ years AI/ML or Automation experience in lieu of degree

Tools

NVIDIA Triton Inference Server

CUDA

cuDNN

TensorRT

ONNX Runtime

Kubernetes

Docker

MLflow

Kubeflow

DVC

Job description

Booz Allen Hamilton seeks an AI/ML engineer to join a team building a converged platform that runs real-time wireless signal processing and AI inference on GPU hardware. You will design, train, and deploy models, build data pipelines, and containerize services on Kubernetes.

The role requires 3+ years with PyTorch/TensorFlow, Python, GPU computing, and the ability to obtain a Secret clearance; a STEM degree with extensive AI/ML experience is required.