Skip to content

Staff AI Engineer

Builds and deploys fine-tuned LLMs and AI agents for real-time voice interactions in consumer lending, ensuring compliance and scalability. Requires 2+ years production ML/AI experience with Python, PyTorch/TensorFlow, and LLM frameworks.

180k – 250kSan Francisco, CAML EngineeringOnsite

About the role

What You'll Do

  • AI Model Integration: Fine-tune and deploy LLMs for real-time voice-based borrower interactions
  • Prompt Engineering: Design and iterate on prompt strategies to optimize AI performance and compliance
  • Evaluate LLMs and AI Agents: Build high quality evals using our proprietary datasets to benchmark AI agent performance and run experiments
  • Custom AI Solutions: Train and implement domain-specific AI models for consumer lending
  • Scalability & Compliance: Ensure AI solutions meet regulatory standards (FDCPA, RESPA, TILA) and scale efficiently
  • Data Pipelines & APIs: Build robust AI-driven workflows that integrate seamlessly with loan servicing platforms
  • Continuous Optimization: Monitor performance metrics and continuously improve agent quality and borrower experience

What We're Looking For

  • 2+ years building and deploying ML/AI systems in production environments
  • Strong proficiency in Python and deep learning frameworks (TensorFlow, PyTorch)
  • Experience with LLM orchestration frameworks (LangChain, LlamaIndex, or similar)
  • Track record of shipping AI products that real users depend on
  • Strong product mindset and ability to translate business requirements into AI solutions
  • Excellent communication skills to collaborate across product, engineering, and compliance teams

Skills

PythonPyTorchTensorFlowLangChainLlamaindexLLMsPrompt EngineeringMachine LearningDeep LearningAI Agents

Similar roles

ML Engineering jobs

Member of Technical Staff - X Search

Develops and operates large-scale search engine infrastructure, including retrieval algorithms, indexing, and ML ranking models integrated with Grok AI. Requires experience with search systems, vector databases, and production ML in Python, Go, or Rust.

180k – 440kPalo Alto, CAML EngineeringOn-siteGoRust

Member of Technical Staff - Post-Training and RL

Develops advanced post-training and reinforcement learning techniques like RLHF/DPO and reward modeling to enhance AI model reasoning, truthfulness, and real-world capabilities at xAI. Seeks passionate AI enthusiasts obsessed with truth-seeking models; prior experience preferred but not required.

180k – 600kPalo Alto, CAML EngineeringOn-siteDpoJAX

Member of Technical Staff - Multimodal Understanding

Develops large-scale distributed systems and pipelines for multimodal AI pre-training, post-training, and inference across image, video, audio, and text. Requires expert Python proficiency, experience with JAX/PyTorch/XLA, and scaling multimodal ML systems.

180k – 440kPalo Alto, CAML EngineeringOn-siteRlJAX

Staff Data Scientist | Modeling

Staff Data Scientist advances ML models for healthcare claims auditing, curating data, developing precise models, and optimizing business impact for health plans. Requires expertise in SQL, Python/R, and building ML models from scratch.

180k – 260kUnited StatesML EngineeringRemoteRSQL

Staff Software Engineer | GenAI & Agentic Workflows

Leads design and development of large-scale AI systems for document processing and agentic workflows in healthcare payment integrity. Requires 6+ years experience with Java/Python, productionizing LLMs/RAG, and distributed systems.

180k – 250kUnited StatesML EngineeringRemote6+ YOERAGRay