AI Engineer

About the Role We're an AI-native company leveraging AI to automate complex back-office workflows and tasks in the freight industry. Our mission is to unlock profits and lower costs by addressing the bad data and inefficient workflows that plague the supply chain. We’re expanding our AI team and are looking for a talented AI Engineer to build both AI models and features that directly impact our business. The role is broad, offering the opportunity to work on everything from training and deploying in-house multimodal Large Language Models (LLMs) to scaling our inference infrastructure and building AI agent workflows. What You'll Work On Your primary focus will be on document extraction and understanding, using multimodal LLMs to extract, normalize, and link data with a high degree of accuracy. You'll also work on scaling our machine learning platform to handle significant increases in training and inference volume. Future projects will include expanding our AI capabilities into workflow automation and audit using AI agents. This role spans multiple domains: ML modeling: Training, evaluating, and deploying models. ML infrastructure: Scaling data infrastructure and improving the reliability of our ML systems. AI engineering: Orchestrating API LLM models to solve business problems. Backend engineering: Building atomic tasks and general backend work in the automation domain. Requirements Qualifications 1-3 years of experience with deep learning frameworks (e.g., PyTorch, TensorFlow). Open to recent graduates with a Computer Science degree from a top university. This is a hybrid role based in our San Francisco office, with the expectation of being in the office 3-5 days per week. Benefits Compensation & Benefits Salary: $130k - $200k Visa sponsorship is not mentioned as available. Potential Projects Scaling the throughput of our inference engine through techniques like continuous batching. Developing methods to fine-tune multimodal LLMs to reduce hallucinations. Building AI agents that audit freight invoices. Due to the high volume of applications we anticipate, we regret that we are unable to provide individual feedback to all candidates. If you do not hear back from us within 4 weeks of your application, please assume that you have not been successful on this occasion. We genuinely appreciate your interest and wish you the best in your job search. Commitment to Equality and Accessibility: At MLabs, we are committed to offer equal opportunities to all candidates. We ensure no discrimination, accessible job adverts, and providing information in accessible formats. Our goal is to foster a diverse, inclusive workplace with equal opportunities for all. If you need any reasonable adjustments during any part of the hiring process or you would like to see the job-advert in an accessible format please let us know at the earliest opportunity by emailing human-resources@mlabs.city. MLabs Ltd collects and processes the personal information you provide such as your contact details, work history, resume, and other relevant data for recruitment purposes only. This information is managed securely in accordance with MLabs Ltd’s Privacy Policy and Information Security Policy, and in compliance with applicable data protection laws. Your data may be shared only with clients and trusted partners where necessary for recruitment purposes. You may request the deletion of your data or withdraw your consent at any time by contacting legal@mlabs.city.