AI Engineer II/ Sr AI Engineer I

We are seeking a talented individuals  to join our cutting-edge team. This position is specifically designed to apply their research expertise in a dynamic industry setting. In this role, you'll tackle complex challenges in large language models (LLMs), optical character recognition (OCR), and model scaling. You'll be at the forefront of developing and optimizing AI systems that push the boundaries of what's possible in machine learning.

• Lead research initiatives to improve OCR accuracy across diverse document types and languages
• Train and fine-tune LLMs using domain-specific data to enhance performance in specialized contexts
• Develop techniques to scale LLMs efficiently for high-volume production environments
• Design and implement novel approaches to model optimization and evaluation
• Collaborate with cross-functional teams to integrate AI solutions into production systems
• Stay current with the latest research and incorporate state-of-the-art techniques
• Document methodologies, experiments, and findings for both technical and non-technical audiences

• PhD. B.Tech, B.E or M.Tech in Computer Science, Machine Learning, AI, or a related field 
• Strong understanding of deep learning architectures, particularly transformer-based models
• Experience with OCR systems and techniques for improving text recognition accuracy
• Proficiency in Python and deep learning frameworks (PyTorch, TensorFlow, or JAX)
• Demonstrated ability to implement and adapt research papers into working code
• Excellent problem-solving skills with a methodical approach to experimentation
• Strong communication skills to explain complex technical concepts clearly

• Research focus in areas relevant to our work (NLP, computer vision, multimodal learning)
• Familiarity with distributed training systems for large-scale models
• Experience with model quantization, pruning, and other efficiency techniques
• Understanding of evaluation methodologies for assessing model performance
• Knowledge of MLOps practices and tools for model deployment
• Publications at top-tier ML conferences (NeurIPS, ICML, ACL, CVPR, etc.)

• Structured onboarding program designed specifically for AI Engineers
• Opportunity to work on frontier AI models with real-world impact
• Access to significant computing resources for ambitious research
• Collaborative environment with other top AI researchers and engineers
• Flexible work arrangements and competitive compensation
• Support for continued professional development and conference attendance
• Clear path for growth into senior technical or leadership roles

Who we are:

FourKites is , the leader in AI-driven supply chain transformation for global enterprises and pioneer of 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 As the leader in AI-driven supply chain transformation, FourKites pioneered the Intelligent Control Tower powered by the world's largest real-time visibility network. Our platform creates comprehensive digital twins of your supply chain with AI-powered digital workers to automate resolution, improve collaboration and drive outcomes across all stakeholders. Unlike traditional control towers, we enable true real-time execution and intelligent fulfillment, transforming both your supply and customer (edited) 

Benefits 

• Medical benefits start on first day of employment
• 36 PTO days( Sick, Casual and Earned) , 5 recharge days, 2 volunteer days 
• Home Office setups and Technology reimbursement
• Lifestyle & Family benefits 
• Annual Swags/ Festive Swags
• Ongoing learning & development opportunities ( Professional development program, Toast Master club etc.)