ML Research Engineer — LLM Training & Cloud Infra (Equity)

Zettafleet; Greater London; On-site; GBP 75,000 - 120,000; Full time

Benefits offered by this job

Equity

Discretionary bonus

28 days + public holidays

Health insurance

Job summary

Zettafleet is hiring a Machine Learning Research Engineer to develop and enhance model training code and the underlying infrastructural software. You will work on pre-training, post-training and evaluation workloads to domain-adapt LLMs with customer data.

The role involves integrating training code with the backend, collaborating with forward-deployed engineers, and owning significant parts of the work with strong autonomy. Equity is offered.

Qualifications

• Experience in industry ML projects including data processing and distributed training.
• Understanding of LLM architectures, pre-training, post-training and evaluation.
• Proficient in Python with PyTorch; familiarity with TensorFlow/JAX is a plus.
• Experience deploying in cloud platforms and with containerisation (Docker).
• Strong CS fundamentals; ability to decompose complex problems into tasks.
• Excellent communication and teamwork skills.

Responsibilities

• Develop and enhance training recipes for large ML models.
• Integrate Python training code with the software infrastructure.
• Collaborate with forward-deployed engineers to capture customer needs.
• Work with autonomy and contribute to best practices and culture.

Skills

Python proficiency

Distributed training

LLMs understanding

Data processing

Problem solving

Collaboration

Tools

PyTorch

Docker

Cloud platforms (AWS/GCP/Azure)

Job description

Zettafleet is hiring a Machine Learning Research Engineer to develop and enhance model training code and the underlying infrastructural software. You will work on pre-training, post-training and evaluation workloads to domain-adapt LLMs with customer data.

The role involves integrating training code with the backend, collaborating with forward-deployed engineers, and owning significant parts of the work with strong autonomy. Equity is offered.