Staff ML Engineer

About AI71:

AI71 is an industry leader in artificial intelligence, delivering innovative solutions that empower developers, businesses and governments to solve complex challenges. AI71 builds secure, enterprise-ready applications powered by cutting-edge technology—tailored for knowledge workers and sector-specific needs. AI71 bridges the gap between advanced AI and real-world impact. Guided by a strong commitment to research and responsibility, we create transformative solutions that drive progress and empower communities. 

The Role:

Are you an experienced Machine Learning Engineer with a passion for AI? Do you have a proven track record of harnessing the power of data, developing advanced algorithms, and building AI models? If so, AI71 offers a unique opportunity for you to play a pivotal role in shaping AI-driven solutions that have the potential to redefine industries. 

As a Machine Learning Engineer at AI71, you’ll be responsible for developing, deploying, and optimizing ML / LLM models that power our AI solutions. You will collaborate with cross-functional teams to translate business challenges into impactful AI products. 

What You'll Do:

• Model Development: Develop, train, and deploy predictive, prescriptive models that enhance the capabilities of our AI solutions. 
• AI Technologies: Work with state-of-the-art AI techniques, including Large Language Models (LLMs) and Retrieval-Augmented Generation (RAG), and apply them to real-world business contexts. 
• Cross-Functional Collaboration: Partner with business stakeholders to understand objectives and translate them into actionable machine learning tasks. 
• Model Optimization: Continuously monitor, evaluate, and improve models based on real-world performance and evolving business needs. 
• Data Pipeline Maintenance: Implement and manage robust data preprocessing pipelines to ensure high-quality, reliable input data for model development. 

What You'll Bring:

• Experience: 7+ years of experience in AI, Machine Learning, Reinforcement Learning (RL), Data Science, or a related field. 
• Expertise in LLMs, Agentic AI: Proven experience with Large Language Models (LLMs) and fine-tuning techniques as well as building agentic workflows. 
• Education: Master’s or Ph.D. in AI, Data Science, Computer Science, Statistics, or a related field. 
• Programming Proficiency: Strong proficiency in Python  
• Preference for candidates with a competitive coding background (e.g., ACM/ICPC, NOI/IOI, Top Coder, Kaggle). 
• Technical Skills: Expertise in machine learning frameworks (e.g., scikit-learn, TensorFlow, PyTorch), Generative AI technologies, and libraries like Langchain, Weaviate, Langgraph, LlamaIndex, AgenticAI workflows. 
• Track Record: Demonstrated success in applying data science and machine learning to solve real-world, business-critical problems. 
• Collaboration: Excellent communication and collaboration skills with the ability to work across interdisciplinary teams. 

Why AI71:

• Mission-Driven Work: Work on cutting-edge AI applications with a talented and passionate team, solving real-world challenges in critical sectors.
• Unparalleled Opportunity: This is a chance to innovate and solve real-world challenges using AI at a company with unique access to world-leading models and resources.
• Career Growth: We offer competitive compensation, benefits, and significant career growth opportunities as a foundational member of the team.
• World-Class Environment: Enjoy a flexible working environment and the latest tools & technologies needed to do your best work.