Founding Lead ML Engineer — Build the Core AI Stack

A1; United States; Remote; USD 120,000 - 150,000; Full time

Job summary

A pioneering AI development company in the United States is seeking an innovative technical lead to shape the core technical direction of their AI application. You will design end-to-end training systems, adapt model architectures, and develop scalable inference frameworks. Ideal candidates will have strong deep learning backgrounds, experience with large model training, and robust software engineering capabilities. This role offers the opportunity to work on impactful AI projects in a collaborative environment.

Qualifications

• Strong background in deep learning and transformer architectures.
• Hands-on experience training or fine-tuning large models (LLMs or vision models).
• Strong software engineering skills—writing robust, production-grade systems.

Responsibilities

• Build end-to-end training pipelines: data → training → eval → inference.
• Design new model architectures or adapt open-source frontier models.
• Architect scalable inference systems using vLLM / TensorRT-LLM / DeepSpeed.

Skills

Deep learning

Transformer architectures

PyTorch

GPU optimization

Software engineering

Tools

TensorFlow

JAX

DeepSpeed

FSDP

Ray

Job description

A pioneering AI development company in the United States is seeking an innovative technical lead to shape the core technical direction of their AI application. You will design end-to-end training systems, adapt model architectures, and develop scalable inference frameworks. Ideal candidates will have strong deep learning backgrounds, experience with large model training, and robust software engineering capabilities. This role offers the opportunity to work on impactful AI projects in a collaborative environment.