Staff AI Platform Engineer II

Summary

Join Kaseya's dynamic team as a Staff AI Platform Engineer II and build the intelligent core of our AI platform. You will engineer autonomous agents that perceive, reason, and learn in complex environments, pioneering agentic design using LLMs, knowledge graphs, and reasoning engines. This 100% remote position requires designing and implementing autonomous AI agents, developing agentic architectures, applying advanced RAG techniques, integrating knowledge graphs, and building robust evaluation frameworks. Collaboration with various teams is essential to ensure seamless deployment and continuous improvement. The ideal candidate possesses extensive experience in applied AI/ML, expertise in Python or TypeScript, and familiarity with various LLMs and agent frameworks.

Requirements

• 5-7+ years of experience in applied AI/ML, with a strong focus on building intelligent systems and autonomous agents
• Deep expertise in Python and/or TypeScript
• Hands-on experience with multiple LLMs (e.g., GPT, Llama, Claude, Gemini), including fine-tuning, advanced RAG techniques, and prompt engineering
• Proficiency with agent frameworks such as LangChain, LangGraph, AutoGen, or CrewAI
• Experience developing agent planning and reasoning algorithms (e.g., Tree of Thoughts, ReAct, Monte Carlo Tree Search)
• Strong background in model evaluation, A/B testing, and versioning for AI models
• Familiarity with vector databases (e.g., Pinecone, Milvus, Chroma)
• Experience with knowledge graphs (e.g., Neo4j, RDF) and symbolic AI approaches
• Understanding of multi-modal LLM capabilities and responsible AI practices
• Experience with explainable AI (XAI) and responsible AI toolkits
• Bachelor's or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field

Responsibilities

• Design and implement autonomous AI agents capable of perception, reasoning, and learning in complex, real-world environments
• Develop and optimize agentic architectures using LLMs, vector databases, and symbolic reasoning engines
• Apply advanced RAG techniques, prompt engineering, and fine-tuning to enhance agent performance
• Integrate knowledge graphs and multi-modal capabilities to enable contextual and adaptive agent behavior
• Build robust evaluation frameworks, A/B testing pipelines, and model versioning systems
• Collaborate with platform, data, and product teams to ensure seamless deployment and continuous improvement of intelligent agents

Preferred Qualifications

• PhD in AI, Cognitive Science, or a related discipline
• Certifications in AI/ML (e.g., DeepLearning.AI, Google Cloud ML Engineer) are a plus
• Contributions to open-source AI projects
• Experience with causal inference and its application in agent decision-making
• Familiarity with formal methods for agent behavior verification
• Background in human-AI interaction design and UX for AI-powered applications
• Experience developing agents for domains like cybersecurity or IT operations automation
• Knowledge of neuro-symbolic AI and reinforcement learning from human feedback (RLHF) or direct preference optimization (DPO)

Benefits

This position is 100% remote