[Hiring] Staff AI Engineer @FourKites

Role Description

We are seeking an experienced Staff AI Engineer to join our AI and Data Platform team, where you'll play a pivotal role in building and scaling our next-generation AI workforce platform. You'll work on cutting-edge agent-based systems that are transforming supply chain operations for Fortune 500 companies, delivering real business value through intelligent automation.

What you'll be doing:

• Technical Leadership:
  • Design and implement production-scale AI agent systems and orchestration frameworks (LangGraph, LangChain, similar architectures)
  • Lead architecture for multi-agent systems handling complex business workflows
  • Optimize deployment strategies using both LLMs and SLMs based on use case requirements
  • Build natural language-configurable business process automation frameworks
  • Implement multi-modal AI systems for document understanding (tables, charts, layouts)
• AI/ML Implementation & Optimization:
  • Deploy and optimize LLMs/SLMs in production with fine-tuning techniques (LoRA, QLoRA, DPO)
  • Implement quantization strategies (INT8, INT4) and model distillation for edge deployment
  • Build evaluation frameworks including LLM-as-judge systems and regression testing
  • Design streaming architectures for real-time LLM responses (SSE, WebSockets)
  • Create semantic caching and embedding-based retrieval systems
  • Develop GraphRAG and long-context handling strategies (100k+ tokens)
• System Architecture & Engineering:
  • Design scalable microservices with comprehensive observability (LangSmith, Arize, custom telemetry)
  • Build secure multi-tenant systems with prompt injection prevention and output validation
  • Implement cost optimization through intelligent model routing and fallback strategies
  • Develop document processing pipelines with OCR and layout understanding
  • Create event-driven architectures for real-time shipment tracking and exception handling
• Data & Infrastructure:
  • Build data pipelines for training data curation, synthetic generation, and PII masking
  • Implement RLHF/RLAIF feedback loops for continuous improvement
  • Design experiment tracking and model registry systems (MLflow, DVC)
  • Optimize inference costs through batch processing and spot instance utilization
  • Establish model governance, audit trails, and compliance frameworks

Qualifications

• 8+ years software engineering, 3+ years in production AI/ML systems
• Expertise in Python, PyTorch/JAX, and AI frameworks (LangChain, Transformers, PEFT)
• Experience with LLMs (GPT-4, Claude, Gemini) and SLMs (Phi, Llama, Mistral)
• Hands-on experience with:
  • Fine-tuning techniques (LoRA, QLoRA, DPO, RLHF)
  • Model optimization (quantization, distillation, pruning)
  • Vector databases and RAG architectures
  • Streaming systems and real-time processing
  • Security measures (prompt injection prevention, jailbreak detection)
• Strong background in distributed systems, Kubernetes, and cloud platforms

Requirements

• Experience with document intelligence and multi-modal AI systems (nice to have)
• Understanding of supply chain operations, EDI/API integrations (nice to have)
• Knowledge of token economics and consumption-based pricing models (nice to have)
• Familiarity with enterprise compliance requirements (GDPR, CCPA, SOC2) (nice to have)

Benefits

• Competitive compensation with stock options
• Outstanding benefits and a collaborative culture for all employees
• 5 global recharge days, in addition to generous PTO and standard holidays
• Parental leave for all parents
• Annual wellness stipend and volunteer days
• Medical benefits start on the first day of employment (for India)
• 36 PTO days (Sick, Casual, and Earned), 5 recharge days, 2 volunteer days (for India)
• Home Office setups and Technology reimbursement (for India)
• Mental Wellness support and guidance (for India)
• Ongoing learning & development opportunities (Professional development program, Toast Master club, etc.) (for India)