Demo

ML & AI Platform Engineer

Nisum
San Jose, CA Full Time
POSTED ON 7/8/2026
AVAILABLE BEFORE 8/6/2026

What You'll Do:

  • Lead end-to-end design, development, deployment, and optimization of Machine Learning and Generative AI solutions.
  • Architect and implement scalable ML and GenAI pipelines on Azure Databricks and enterprise cloud platforms.
  • Translate business problems into robust machine learning solutions through data exploration, feature engineering, model development, and optimization.
  • Build and operationalize advanced AI applications leveraging Large Language Models (LLMs), RAG architectures, and agentic AI frameworks.
  • Develop scalable inference and model serving solutions for batch and real-time workloads.
  • Deploy, monitor, and maintain ML and GenAI systems with a focus on reliability, scalability, observability, and performance.
  • Collaborate with Data Engineering, Product, and business stakeholders to deliver impactful AI solutions.
  • Drive adoption of MLOps and LLMOps best practices including CI/CD automation, testing, versioning, and lifecycle management.
  • Implement and optimize vector search, embeddings, semantic retrieval, and enterprise AI workflows.
  • Mentor junior engineers and contribute to technical excellence across the AI organization.
  • Ensure security, governance, compliance, and cost-effective operation of AI solutions in production.
  • Strong ownership mindset with the ability to independently solve complex modeling challenges.
  • Ability to translate business requirements into scalable AI and machine learning solutions.

What You Know:

  • 5 years of experience in Machine Learning Engineering, Data Science, or AI Engineering roles.
  • Proven track record of hands-on model building and solving real-world machine learning challenges independently.
  • Advanced proficiency in Python and machine learning frameworks including Scikit-learn, PyTorch, and TensorFlow.
  • Deep expertise in Azure Databricks and Azure-based machine learning and data services.
  • Strong experience designing and deploying machine learning models from development through production.
  • Hands-on experience with Generative AI applications, LLMs, RAG systems, and agentic AI frameworks.
  • Experience with orchestration frameworks such as LangChain, LangGraph, Semantic Kernel, or similar technologies.
  • Familiarity with vector databases such as Pinecone, Weaviate, FAISS, Milvus, or Azure AI Search.
  • Experience with embeddings, semantic search, retrieval pipelines, and knowledge-grounded AI applications.
  • Strong understanding of model serving, inference pipelines, monitoring, and observability for AI systems.
  • Experience deploying and maintaining AI/ML solutions in enterprise production environments.
  • Knowledge of MLOps practices including CI/CD pipelines, model versioning, testing, and automation.
  • Familiarity with containerization and deployment technologies including Docker, Kubernetes, and REST APIs.
  • Experience working with GPU-accelerated machine learning and Generative AI workloads.
  • Exceptional problem-solving, communication, and stakeholder management skills.

Education:

  • Bachelor's degree in Computer Science, Engineering, Data Science, Statistics, Mathematics, or related field (required).
  • Master's degree is a plus.

Benefits:

  • In addition to competitive salaries and benefits packages, Nisum US offers its employees some unique and fun extras:
  • Professional Development - We offer in-house technical training and professional learning programs aimed at developing skills across a broad spectrum of topics such as technology, leadership, role-based training, and process expertise. We also offer an annual stipend for employees to attend external courses in order to maintain professional certifications.
  • Health & Wellness Benefits - We believe that your health and welfare are important, and we strive to ensure that you have affordable options available to you, including some plans that are subsidized for employees and their families up to 90%. We also have dental and vision plans in the US where Nisum pays 100% of premiums for employees.
  • Volunteerism Pay - We believe in giving back and in the US, our employees are eligible for up to 40 hours of paid time off each year to volunteer towards the causes that they are most passionate about. This is in addition to personal PTO and paid holidays.
  • Additional Benefits - We offer all the other important benefits to keep employees and their families healthy and financially secure, such as 401(k) retirement savings with a company match, pre-tax parking and transit programs, disability insurance, and Basic Life/AD&D, alongside exclusive employee discounts on a wide variety of products and services.

Interview Process:

  • Shortlisted candidates will be evaluated on coding skills through a hands-on technical exercise during the interview process.

Compensation Band:

  • $140K - $150K per annum

Salary : $140,000 - $150,000

If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

What is the career path for a ML & AI Platform Engineer?

Sign up to receive alerts about other jobs on the ML & AI Platform Engineer career path by checking the boxes next to the positions that interest you.
Income Estimation: 
$101,387 - $124,118
Income Estimation: 
$119,030 - $151,900
Income Estimation: 
$119,030 - $151,900
Income Estimation: 
$149,493 - $192,976
Employees: Get a Salary Increase
View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

Job openings at Nisum

  • Nisum Marlborough, MA
  • Nisum is a leading global digital commerce firm headquartered in California, with services spanning digital strategy and transformation, insights and analy... more
  • 6 Days Ago

  • Nisum San Francisco, CA
  • Role Overview The Global Product team is seeking a Product Manager to support delivery of platform capabilities across international markets, including com... more
  • 2 Days Ago


Not the job you're looking for? Here are some other ML & AI Platform Engineer jobs in the San Jose, CA area that may be a better fit.

  • techire ai Fremont, CA
  • ML Platform Engineer / ML Infrastructure Engineer Want to build the systems that determine how quickly frontier AI research moves? The biggest bottleneck i... more
  • 5 Days Ago

  • Advanced Micro Devices, Inc Santa Clara, CA
  • WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data... more
  • 4 Days Ago

AI Assistant is available now!

Feel free to start your new journey!