Senior AI/ML Engineer - Remote, Production-Scale ML

Prosum; California; Remote; USD 140,000 - 180,000; Full time

Job summary

Prosum is seeking a Senior Software Engineer specializing in AI/ML to join a pioneering AI engineering organization. This role focuses on building, training, evaluating, and deploying AI/ML models and agentic systems that power intelligent products in enterprise applications.

You will apply PyTorch, transformer architectures, and LLM expertise across the full ML lifecycle, ensuring production-grade reliability and scalability in a large-scale environment.

Qualifications

• 6-10+ years of professional software engineering experience.
• 3+ years focused on production ML/AI engineering.
• Expert-level proficiency in Python and strong familiarity with the Python ML/AI ecosystem.
• Hands-on production experience with PyTorch, including model development and CUDA acceleration.
• Experience with Hugging Face Transformers, Datasets, and PEFT or comparable libraries.
• Demonstrated experience building and evaluating RAG pipelines with vector stores.
• Production experience integrating LLM APIs and building reliable prompt engineering systems.
• Experience with LangChain, LangGraph for agentic and tool-calling workflows.

Responsibilities

• Design, develop, train, and evaluate machine learning models using PyTorch.
• Build ML training pipelines, experiment tracking workflows, and evaluation frameworks.
• Implement transformer-based models and LLM integrations for production use cases.
• Apply parameter-efficient fine-tuning techniques for domain adaptation.
• Design RAG architectures with vector databases and semantic search.
• Optimize inference for latency and throughput in production.
• Develop AI evaluation frameworks and automated tests for reliability.

Skills

Python

PyTorch

LLMs

NLP

Transformer architectures

RAG pipelines

LangChain

LangGraph

ML lifecycle

Education

Bachelor’s or Master’s degree in Computer Science / ML / Statistics

Tools

Docker

Kubernetes

MLflow

Weights & Biases

Comet

Hugging Face Transformers

Datasets

LangChain

LangGraph

AWS SageMaker

Azure ML

Job description

Prosum is seeking a Senior Software Engineer specializing in AI/ML to join a pioneering AI engineering organization. This role focuses on building, training, evaluating, and deploying AI/ML models and agentic systems that power intelligent products in enterprise applications.

You will apply PyTorch, transformer architectures, and LLM expertise across the full ML lifecycle, ensuring production-grade reliability and scalability in a large-scale environment.