Senior AI/ML Engineer - Production-Ready PyTorch & LLMs

Prosum; California (MO); Hybrid; USD 140,000 - 180,000; Full time

Job summary

Prosum seeks a Senior Software Engineer specializing in AI and Machine Learning to join a leading AI engineering team. This role sits at the intersection of applied ML, LLM product development, and production-grade software engineering.

You will build, train, evaluate, and deploy AI/ML models and agentic systems powering intelligent products across enterprise apps, combining deep hands-on expertise with robust engineering discipline to deliver scalable AI solutions.

Qualifications

• Bachelor's or Master's in a quantitative field.
• 6–10+ years of professional software engineering experience.
• 3+ years focused on production ML/AI engineering.
• Expert-level Python and ML/AI ecosystem familiarity.
• Hands-on PyTorch experience with GPU acceleration and model serialization.
• Experience with Transformers, Datasets, PEFT, and RAG pipelines.
• Experience integrating LLM APIs and building prompt engineering systems.
• Experience with LangChain/LangGraph or equivalent for agentic workflows.
• Containerization and cloud ML platforms; Docker, Kubernetes, AWS SageMaker, Azure ML.

Responsibilities

• Design, develop, train, fine-tune, and evaluate ML models with PyTorch.
• Build ML pipelines, experiment tracking, and evaluation frameworks.
• Implement transformer-based models and LLM integrations for production.
• Apply PEFT techniques to adapt models to domain needs.
• Design RAG architectures with vector databases and semantic search.
• Optimize inference latency and throughput via quantization and batching.
• Develop AI evaluation frameworks ensuring safe, production-ready models.
• Mentor engineers and promote best practices for reproducible ML.

Skills

Python

PyTorch

Transformer architectures

LLMs

RAG pipelines

LangChain

LangGraph

PEFT

Docker

Kubernetes

MLflow

Weights & Biases

Comet

SageMaker

Azure ML

Education

Bachelor's or Master's degree in Computer Science, Machine Learning, Statistics, Mathematics, or a related quantitative field

Tools

TensorFlow

JAX

ONNX

Triton Inference Server

Job description

Prosum seeks a Senior Software Engineer specializing in AI and Machine Learning to join a leading AI engineering team. This role sits at the intersection of applied ML, LLM product development, and production-grade software engineering.

You will build, train, evaluate, and deploy AI/ML models and agentic systems powering intelligent products across enterprise apps, combining deep hands-on expertise with robust engineering discipline to deliver scalable AI solutions.