Jobs, fresher vacancy - Machine Learning Engineer, Platform • UK

Company: AION

UK

About AIONAION is building an interoperable AI cloud platform by transforming the future of high-performance computing (HPC) through its decentralized AI cloud. Purpose-built for bare-metal performance, AION democratizes access to compute and provides managed services, aiming to be an end-to-end AI lifecycle platform—taking organizations from data to deployed models using its forward-deployed engineering approach.AI is transforming every business around the world, and the demand for compute is surging like never before. AION thrives to be the gateway for dynamic compute workloads by building integration bridges with diverse data centers around the world and re-inventing the compute stack via its state-of-the-art serverless technology. We stand at the crossroads where enterprises are finding it hard to balance AI adoption with security. At AION, we take enterprise security and compliance very seriously and are re-thinking every piece of infrastructure from hardware and network packets to API interfaces.Led by high-pedigree founders with previous exits, AION is well-funded by major VCs with strategic global partnerships. Headquartered in the US with global presence, the company is building its initial core team in India/UK.Who You AreYou're a hands-on ML engineer with 4-6 years of experience building and fine-tuning large language models (LLMs) and transformer-based models. You're execution-focused and thrive on solving challenging problems at the intersection of machine learning research and production systems.You're comfortable working across the ML development lifecycle—from data preparation and model fine-tuning to evaluation and optimization. You understand both what makes a model perform well and how to systematically improve model quality through experimentation. Experience with LLM fine-tuning (LoRA, QLoRA), RLHF pipelines, and comprehensive model evaluation is highly desirable. You bring strong ownership, initiative, and the drive to build production-ready ML models that impact thousands of developers globally.RequirementsWhat You'll DoML Model Development & OptimizationDesign and implement end-to-end LLMOps pipelines for model training, fine-tuning, and evaluationFine-tune and customize LLMs (Llama, Mistral, Gemma, etc.) using full fine-tuning and PEFT techniques (LoRA, QLoRA) with tools like Unsloth, Axolotl, and HuggingFace TransformersImplement RLHF (Reinforcement Learning from Human Feedback) pipelines for model alignment and preference optimizationDesign experiments for automated hyperparameter tuning, training strategies, and model selectionPrepare and validate training datasets—ensuring data quality, preprocessing, and format correctnessBuild comprehensive model evaluation systems with custom metrics (BLEU, ROUGE, perplexity, accuracy) and develop synthetic data generation pipelinesOptimize model accuracy, token efficiency, and training performance through systematic experimentationDesign and maintain prompt engineering workflows with version control systemsDeploy models using vLLM with multi-adapter LoRA serving, hot-swapping, and basic optimizations (speculative decoding, continuous batching, KV cache management)ML Operations & Technical LeadershipSet up ML-specific monitoring for model quality, drift detection, and performance tracking with automated retraining triggersManage model versioning, artifact storage, lineage tracking, and reproducibility using experiment tracking toolsDebug production model issues and optimize cost-performance trade-offs for training and inferencePartner with infrastructure engineers on ML-specific compute requirements and deployment pipelinesDocument model development processes and share knowledge through internal tech talksTechnical Skills & ExperienceIf you are meeting some of these requirements and feel comfortable catching up on others, we definitely recommend you to apply:4-6 years of hands-on experience in machine learning engineering or applied ML rolesStrong fine-tuning experience with modern LLMs—practical knowledge of transformer architectures, attention mechanisms, and both full fine-tuning and PEFT techniques (LoRA/QLoRA)Deep understanding of transformer model architectures including modern variants (MoE, Grouped-Query Attention, Flash Attention, state space models)Production ML experience—you've built and fine-tuned models for real-world applicationsProficiency in Python and ML frameworks (PyTorch, HuggingFace Transformers, PEFT, TRL) with hands-on experience in tools like Unsloth and AxolotlExperience building model evaluation systems with metrics like BLEU, ROUGE, perplexity, and accuracyHands-on experience with prompt engineering, synthetic data generation, and data preprocessing pipelinesBasic deployment experience with vLLM including multi-adapter serving, hot-swapping, and inference optimizationsUnderstanding of GPU computing—memory management, multi-GPU training, mixed precision, gradient accumulationStrong debugging skills for training failures, OOM errors, convergence issues, and data quality problemsExperience with model alignment techniques (RLHF, DPO) and implementing RLHF pipelines is highly desirableExperience with distributed training (DeepSpeed, FSDP, DDP) is a plusKnowledge of model quantization techniques (GPTQ, AWQ) and their impact on model quality is desirablePrior experience with AWS SageMaker, MLflow for experiment tracking, and Weights & Biases is a strong plusExposure to cloud platforms (AWS/GCP/Azure) for training workloads is beneficialFamiliarity with Docker containerization for reproducible training environmentsPreferred AttributesHigh ownership, self driven and bias for action.Strong strategic thinking and ability to connect technical decisions to business impact.Excellent communication and mentoring skills.Thrives in ambiguity, fast-paced environments, and early-stage startup culture.BenefitsWhy Join AION?Work directly with high-pedigree founders shaping technical and product strategy.Build infrastructure powering the future of AI compute globally.Significant ownership and impact with equity reflective of your contributions.Competitive compensation, flexible work options, and wellness benefits.Apply Now:If you’re a machine learning engineer ready to lead MLAAS(Machine learning as a Service) architecture and scale next-generation AI infrastructure, we want to hear from you. Please share the following in the summary section:Your resume highlights relevant projects and leadership experienceLinks to products, code(Github), or demos you’ve built.A brief note on why AION’s mission excites you.