KLA

AI Software Engineer - Operations

KLA
Apply
24 hours ago
Ann Arbor, MI, USAMid Level
H1B sponsor

Base Salary

$90k - $133k/yr

Responsibilities

  • Design and develop end-to-end AI applications that leverage LLMs for business use cases.
  • Build and maintain backend services that orchestrate LLM interactions, prompts, and workflows.
  • Prototype and develop intuitive user interfaces for human-AI interaction.
  • Integrate LLM APIs such as OpenAI, Azure OpenAI, and Anthropic into scalable applications.
  • Develop prompt engineering strategies and manage the prompt lifecycle.
  • Implement retrieval-augmented generation architectures using internal and external data sources.
  • Design data pipelines and connectors to databases, APIs, knowledge bases, and enterprise systems.
  • Monitor, evaluate, and improve AI application performance using latency, cost, and response-quality metrics.
  • Collaborate with product managers, data scientists, and business stakeholders to deliver AI-enabled solutions.
  • Apply responsible AI practices addressing privacy, security, and bias mitigation.
  • Rapidly prototype AI tools and iterate based on user feedback.
  • Lead and evolve global electrical engineering standards, review facility electrical designs, and guide technical strategy for critical power infrastructure.
  • Support renewable energy and electrification initiatives including solar, battery storage, EV charging, and utility coordination.

Requirements

  • Bachelor’s degree in Computer Science, Software Engineering, or a related field.
  • At least three years of professional software development experience.
  • Strong programming skills in Python, JavaScript, or similar languages.
  • Experience building AI products with LLM frameworks such as LangChain, Semantic Kernel, or LlamaIndex.
  • Knowledge of RAG and vector databases such as Pinecone, FAISS, Azure Cognitive Search, and Milvus.
  • Experience with Neo4j, knowledge graphs, enterprise data platforms, and integrations such as Snowflake, SAP, and APIs.
  • Experience orchestrating multi-step AI workflows or agent-based systems.
  • Experience designing conversational interfaces or chat-based applications.
  • Understanding of LLM output evaluation, grounding, and hallucination mitigation.
  • Exposure to prompt optimization and fine-tuning concepts.
  • Experience deploying scalable microservices architectures.
  • Strong UI/UX sensibility for AI-driven applications.
  • Experience delivering production-grade AI solutions in cross-functional teams.
  • Experience developing web-based user interfaces with React, Angular, or similar frameworks.
  • Experience working in or with a global manufacturing company.

Benefits

  • Base pay is listed at $90,400.00–$132,600.00 annually.
  • Benefits may include medical, dental, vision, life, voluntary benefits, 401(k) with company matching, ESPP, student debt assistance, tuition reimbursement, career development, financial planning, wellness programs, paid time off, paid holidays, and family care and bonding leave.
  • The primary location is KLA’s Ann Arbor, Michigan site in the United States.

Tech Stack

AngularJavaScriptNeo4jPythonReactSnowflake
KLA

About KLA

10,000+ employees

KLA builds process control, inspection and metrology systems used to manufacture semiconductor wafers, reticles, advanced packaging, printed circuit boards, and displays, and sells related software and services to chipmakers and electronics manufacturers. Its business model centers on capital equipment sales plus service and support contracts. Headquartered in Milpitas, California, KLA traces its origins to the 1997 KLA-Tencor merger and is publicly traded on NASDAQ under the ticker KLAC.

Contact me