AI Engineering Intern

Responsibilities

• Conduct original research on LLM agent architectures and optimization techniques
• Design and implement intelligent agent architectures for complex enterprise automation tasks
• Develop novel methods for parameter-efficient adaptation, alignment, and reinforcement learning
• Optimize inference pipelines through systems-level innovations and kernel development
• Develop and evaluate novel algorithms with academic rigor and production feasibility
• Present work at internal research seminars and external conferences
• Mentor and collaborate with LLM engineers on implementation and deployment

Requirements

• Currently pursuing MS/PhD in Computer Science, Machine Learning, NLP, or related fields
• Publications at top-tier venues (ICML, NeurIPS, ICLR, ACL, EMNLP, NAACL)
• Strong programming skills in Python and PyTorch
• Ability to work in-person at the San Francisco office
• Ability to work independently and collaborate across teams

Preferred Qualifications

• Experience with self-evolving agent systems
• Proficiency in CUDA programming and custom kernel development
• Background in reinforcement learning-based LLM fine-tuning
• Contributions to production inference systems (vLLM, TensorRT-LLM, SGLang, or Hugging Face)
• Experience bridging academic research with production systems
• Open-source contributions to ML infrastructure projects

About the Company

Workato delivers enterprise infrastructure for the agentic era, helping enterprises unify data, applications, processes, and AI into a single, governed platform. The Workato AI Lab develops cutting-edge agentic systems that transform how businesses automate and optimize their workflows.