Senior ML Architect for Production-Scale Transformers

AI Chopping Block; Northern (KY); Hybrid; USD 150,000 - 230,000; Full time

Job summary

A1 is seeking a Principal Machine Learning Engineer to serve as a technical authority for the most critical ML systems. You will oversee design, training, evaluation, and infrastructure, setting standards that scale across the organization.

The role is hands-on and high impact, balancing latency, throughput, cost, and reliability while collaborating with backend, mobile, and desktop teams to ship robust AI-driven features.

Qualifications

• Strong background in deep learning and transformer-based architectures.

Responsibilities

• Architect and build large-scale ML systems spanning data, training, evaluation, inference, and deployment.

Skills

Deep learning

Transformer models

Production ML

PyTorch

Distributed training

GPU optimization

Software engineering

Ambiguity handling

Tools

DeepSpeed

Ray

TensorRT

Megatron

Job description

A1 is seeking a Principal Machine Learning Engineer to serve as a technical authority for the most critical ML systems. You will oversee design, training, evaluation, and infrastructure, setting standards that scale across the organization.

The role is hands-on and high impact, balancing latency, throughput, cost, and reliability while collaborating with backend, mobile, and desktop teams to ship robust AI-driven features.