FourKites

<html><body><div class="job__description body">  <div><h2><strong>About the Role</strong></h2><p>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.</p><h2><strong>Key Responsibilities</strong></h2><h3><strong>Technical Leadership</strong></h3><ul><li>Design and implement production-scale AI agent systems and orchestration frameworks (LangGraph, LangChain, similar architectures)</li><li>Lead architecture for multi-agent systems handling complex business workflows</li><li>Optimize deployment strategies using both LLMs and SLMs based on use case requirements</li><li>Build natural language-configurable business process automation frameworks</li><li>Implement multi-modal AI systems for document understanding (tables, charts, layouts)</li></ul><h3><strong>AI/ML Implementation & Optimization</strong></h3><ul><li>Deploy and optimize LLMs/SLMs in production with fine-tuning techniques (LoRA, QLoRA, DPO)</li><li>Implement quantization strategies (INT8, INT4) and model distillation for edge deployment</li><li>Build evaluation frameworks including LLM-as-judge systems and regression testing</li><li>Design streaming architectures for real-time LLM responses (SSE, WebSockets)</li><li>Create semantic caching and embedding-based retrieval systems</li><li>Develop GraphRAG and long-context handling strategies (100k+ tokens)</li></ul><h3><strong>System Architecture & Engineering</strong></h3><ul><li>Design scalable microservices with comprehensive observability (LangSmith, Arize, custom telemetry)</li><li>Build secure multi-tenant systems with prompt injection prevention and output validation</li><li>Implement cost optimization through intelligent model routing and fallback strategies</li><li>Develop document processing pipelines with OCR and layout understanding</li><li>Create event-driven architectures for real-time shipment tracking and exception handling</li></ul><h3><strong>Data & Infrastructure</strong></h3><ul><li>Build data pipelines for training data curation, synthetic generation, and PII masking</li><li>Implement RLHF/RLAIF feedback loops for continuous improvement</li><li>Design experiment tracking and model registry systems (MLflow, DVC)</li><li>Optimize inference costs through batch processing and spot instance utilization</li><li>Establish model governance, audit trails, and compliance frameworks</li></ul><h2><strong>Required Qualifications</strong></h2><h3><strong>Technical Skills</strong></h3><ul><li>8+ years software engineering, 3+ years in production AI/ML systems</li><li>Expertise in Python, PyTorch/JAX, and AI frameworks (LangChain, Transformers, PEFT)</li><li>Experience with LLMs (GPT-4, Claude, Gemini) and SLMs (Phi, Llama, Mistral)</li><li>Hands-on experience with:</li><ul><li>Fine-tuning techniques (LoRA, QLoRA, DPO, RLHF)</li><li>Model optimization (quantization, distillation, pruning)</li><li>Vector databases and RAG architectures</li><li>Streaming systems and real-time processing</li><li>Security measures (prompt injection prevention, jailbreak detection)</li></ul><li>Strong background in distributed systems, Kubernetes, and cloud platforms</li></ul><h3><strong>Domain Knowledge(nice to have)</strong></h3><ul><li>Experience with document intelligence and multi-modal AI systems</li><li>Understanding of supply chain operations, EDI/API integrations</li><li>Knowledge of token economics and consumption-based pricing models</li><li>Familiarity with enterprise compliance requirements (GDPR, CCPA, SOC2)</li></ul><h3><strong>Professional Skills</strong></h3><ul><li>Track record of delivering complex projects with measurable business impact</li><li>Experience with technical sales support, POCs, and customer success</li><li>Strong communication for technical and non-technical audiences</li><li>Data-driven decision making for model selection and cost optimization</li></ul><h2><strong>Preferred Qualifications</strong></h2><ul><li>Supply chain, logistics, or transportation management experience</li><li>Experience with OCR pipelines and document extraction at scale</li><li>Knowledge of GraphRAG and knowledge graph integration</li><li>Contributions to open-source AI projects (Hugging Face, Ollama)</li><li>Experience reducing inference costs by 50%+ through optimization</li><li>Familiarity with MoE architectures and constitutional AI approaches</li><li>Background in building usage-based billing and margin optimization</li><li>Experience with specialized tools (vLLM, TGI, Triton, ONNX, TensorRT)</li></ul><h2><strong>What You'll Work On</strong></h2><ul><li>Building specialized AI agents solving supply chain problems</li><li>Fine-tuning domain-specific models for supply chain terminology</li><li>Implementing hybrid architectures combining cloud LLMs with edge SLMs</li><li>Creating secure document intelligence systems for Fortune 500 clients</li><li>Developing real-time exception handling for shipment tracking</li><li>Building observability and evaluation frameworks for agent performance</li><li>Designing fallback strategies and multi-provider redundancy</li></ul><h2><strong>Technical Environment</strong></h2><p><strong>Models</strong>: GPT-4, Claude, Gemini, Llama 3, Mistral, Phi-3, custom fine-tuned models<br/><strong>Fine-tuning</strong>: LoRA/QLoRA, PEFT, DeepSpeed, bitsandbytes, Axolotl<br/><strong>Infrastructure</strong>: Kubernetes, AWS SageMaker/Bedrock, GPU clusters, edge devices<br/><strong>Frameworks</strong>: LangChain, LangGraph, vLLM, FastAPI, Transformers<br/><strong>Observability</strong>: LangSmith, Weights & Biases, custom telemetry<br/><strong>Data</strong>: PostgreSQL, Redis, Vector DBs, Kafka, feature stores</p><h2><strong>Impact & Growth</strong></h2><p>You'll directly contribute to AI initiatives generating millions in revenue while shaping systems processing millions of transactions daily. Lead technical decisions affecting 25+ engineers while mentoring the next generation of AI engineers. Be at the forefront of production AI optimization, balancing performance, cost, and latency for enterprise customers.</p><p><strong>Who we are:<br/><br/></strong>FourKites®, the leader in AI-driven supply chain transformation for global enterprises and pioneer of advanced real-time visibility, turns supply chain data into automated action. FourKites’ Intelligent Control Tower™ breaks down enterprise silos by creating a real-time digital twin of orders, shipments, inventory and assets. This comprehensive view, combined with AI-powered digital workers, enables companies to prevent disruptions, automate routine tasks, and optimize performance across their supply chain. FourKites processes over 3.2 million supply chain events daily — from purchase orders to final delivery — helping 1,600+ global brands prevent disruptions, make faster decisions and move from reactive tracking to proactive supply chain orchestration.<br/><strong><br/>Working at FourKites<br/></strong><br/>We provide competitive compensation with stock options, outstanding benefits and a collaborative culture for all employees around the globe, including:</p><div><ul><li>5 global recharge days, in addition to standard holidays, and a hybrid, flexible approach to work.</li><li>Parental leave for all parents, an annual wellness stipend and volunteer days also provide you with time and resources for self care and to care for others.</li><li>Opportunities throughout the year to learn and celebrate diversity.</li><li>Access to leading AI tools and foundation models, with the freedom to experiment and find creative ways to be more effective in your role</li></ul>And we're always listening for new ways to support everyone in and out of the office.</div></div></div></body></html>

Job Details

Skills Required

More Details

About the Role

Key Responsibilit...

Apply