Digital Waffle

machine learning engineer for an AI agent

In short

Senior machine learning engineer role building the ML layer of an AI agent product that performs tasks, manages workflows, and retains context across conversations. The role covers end-to-end ML pipelines, fine-tuning large models, inference architecture, and production deployment with GPU optimisation. Requires deep transformer/deep learning knowledge, hands-on production experience training and shipping large-scale models, and proficiency in a major ML framework like PyTorch or JAX. Fully remote, full-time position.

Описание
The company is building an AI agent that performs everyday tasks, manages workflows, and retains context across long conversations. Its machine learning layer is a core part of the product.
Задачи
•Build end-to-end pipelines for data, training, evaluation, and inference
•Adapt and fine-tune models using LoRA, QLoRA, SFT, DPO, and distillation
•Architect inference systems that meet latency and cost constraints
•Create pipelines for high-quality synthetic and real-world training data
•Evaluate robustness, safety, bias, and production behaviour beyond benchmarks
•Own model deployment, including GPU optimisation, quantisation, memory efficiency, and scaling
•Work with application engineers to integrate ML into backend, mobile, and desktop products
•Take ownership of ambiguous problems from zero to one
•Ship, iterate, and learn from production
•Influence how the architecture evolves
Требования
•Deep understanding of deep learning and transformer architectures
•Proven experience training, fine-tuning, or shipping large-scale models in production
•Strong knowledge of at least one major ML framework, such as PyTorch or JAX, and ability to learn others quickly
•Familiarity with distributed training and inference tools, including DeepSpeed, FSDP, Megatron, ZeRO, or Ray
•Engineering discipline in writing readable, robust, and maintainable code
•Experience optimising for GPU constraints, including quantisation, mixed precision, and memory
Nice to have:
•LLM inference frameworks such as vLLM, TensorRT-LLM, or FasterTransformer
•RLHF methods such as PPO, DPO, or ORPO
•Open-source contributions to ML or systems libraries
•Scientific computing, compiler, or GPU kernel experience
•Multimodal or diffusion model experience
•Large-scale data processing with Arrow, Spark, or Ray
Условия
No conditions specified

Stop applying.
Get referred.

$10 in free credits

In just 60 seconds, fully on autopilot — Eraldt's AI agents read your CV, map the roles you fit, find live vacancies and the people who can refer you, write personal emails, and get you referred.

This isn't an application: your CV doesn't go to the company, and we don't promise a referral for this exact role.

Discover who else lands interviews on autopilot