AI Tech Lead - US

StradIT; Jersey City (NJ); Hybrid; USD 190,000 - 230,000; Full time

Job summary

StradIT in Jersey City is seeking an experienced AI Engineering Lead to design, build, and scale advanced AI systems from model development through production deployment. You will lead the architecture of ML solutions and define engineering best practices.

The role requires hands‑on Python experience and proficiency with PyTorch, TensorFlow, or JAX, plus cloud platforms and Docker/Kubernetes. You will mentor engineers and collaborate with research, product, and engineering teams to deliver

Qualifications

• 8+ years of software engineering experience with 2+ years in leading or mentoring teams.
• Strong background in machine learning systems, data pipelines, or distributed computing.
• Proficiency in Python (preferred), plus frameworks such as PyTorch, TensorFlow, or JAX.

Responsibilities

• Lead the architecture and implementation of AI systems from model development to deployment.
• Define engineering best practices for AI workflows, model lifecycle, and system reliability.
• Mentor engineers and drive technical excellence across the organization.
• Collaborate with research, product, and engineering teams to deliver high‑impact AI solutions.

Skills

Leadership experience

Communication skills

Team mentoring

Problem solving

Tools

Python

PyTorch

TensorFlow

JAX

Docker

Kubernetes

MLflow

Kubeflow

Airflow

AWS

GCP

Azure

Job description

We are seeking an experienced AI Engineering Lead to design, build, and scale advanced AI systems from model development through production deployment. The role involves leading the architecture of machine learning solutions, defining engineering best practices, and ensuring robust, reliable, and scalable AI workflows.

The ideal candidate has strong expertise in machine learning systems, data pipelines, and distributed computing, along with hands‑on experience in Python and frameworks such as PyTorch, TensorFlow, or JAX. This role requires experience with cloud platforms (AWS, GCP, or Azure), containerization tools like Docker and Kubernetes, and familiarity with MLOps practices including model versioning, deployment, and monitoring.

You will collaborate closely with research, product, and engineering teams to deliver high‑impact AI solutions, while also mentoring engineers and driving technical excellence across the organization. Strong leadership and communication skills are essential, along with a proven ability to bridge research and production environments.

This is a hybrid position for candidates located in NY/NJ that can commute to the office 4 days a week.

Required Qualifications

• Lead the architecture and implementation of AI systems, from model development to deployment.
• Define engineering best practices for AI workflows, model lifecycle, and system reliability.
• 8+ years of experience in software engineering, including 2+ years leading or mentoring teams.
• Strong background in machine learning systems, data pipelines, or distributed computing.
• Proficiency in Python (preferred), plus experience with frameworks such as PyTorch, TensorFlow, or JAX.
• Experience with cloud platforms (AWS, GCP, or Azure) and container orchestration (Kubernetes, Docker).
• Familiarity with MLOps tools (MLflow, Kubeflow, Airflow, etc.).
• Strong understanding of model deployment, versioning, and monitoring.
• Excellent communication skills and ability to collaborate with research and product stakeholders.