companyAION logo

Machine Learning Engineer, Platform

AIONLondon, England, United Kingdom
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Mid to Senior

Qualifications

What You'll DoML Model Development & OptimizationDesign and implement end-to-end LLMOps pipelines for model training, fine-tuning, and evaluation. Fine-tune and customize LLMs (Llama, Mistral, Gemma, etc.) utilizing full fine-tuning and PEFT techniques (LoRA, QLoRA) with tools such as Unsloth, Axolotl, and HuggingFace Transformers. Implement RLHF (Reinforcement Learning from Human Feedback) pipelines for model alignment and preference optimization. Design experiments aimed at automating hyperparameter tuning, training strategies, and model selection. Prepare and validate training datasets to ensure data quality, preprocessing, and correct formatting. Develop comprehensive model evaluation systems with custom metrics (BLEU, ROUGE, perplexity, accuracy) and create synthetic data for testing.

About the job

About AION

AION is pioneering an interoperable AI cloud platform that is set to revolutionize high-performance computing (HPC) through its innovative decentralized AI cloud. Designed specifically for bare-metal performance, AION democratizes access to computational resources and delivers managed services, positioning itself as a comprehensive AI lifecycle platform—facilitating organizations from raw data to deployed models through its forward-deployed engineering approach.

The global business landscape is undergoing a seismic transformation due to AI, with a skyrocketing demand for computational power. AION aspires to be the gateway for dynamic compute workloads by forging integration bridges with various data centers globally and re-engineering the compute stack with its cutting-edge serverless technology. We recognize the challenges enterprises face in balancing AI adoption with security needs and are committed to ensuring robust enterprise security and compliance by redesigning every aspect of our infrastructure, from hardware and network packets to API interfaces.

With a leadership team that boasts high-caliber founders and significant venture capital backing, AION is establishing its core team across India and the UK while being headquartered in the US.

Who You Are

You are a proactive ML engineer with 4-6 years of experience in developing and fine-tuning large language models (LLMs) and transformer-based architectures. You possess a strong focus on execution and enjoy tackling complex challenges at the intersection of machine learning research and production systems.

Your expertise spans the entire ML development lifecycle—from data preparation and model fine-tuning to rigorous evaluation and optimization. You have a deep understanding of what drives model performance and how to systematically enhance model quality through experimentation. Familiarity with LLM fine-tuning (LoRA, QLoRA), RLHF pipelines, and comprehensive model evaluation is highly desirable. You exhibit strong ownership and initiative, coupled with a passion for developing production-ready ML models that have a global impact on thousands of developers.

About AION

AION is at the forefront of creating an innovative AI cloud platform that enhances high-performance computing through decentralized solutions. With a commitment to security and compliance, AION is reshaping the infrastructure landscape while expanding its global reach and building a robust core team.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.