Training Infrastructure Engineer — Large-Scale AI Systems

Inherentlabs; Greater London; On-site; GBP 70,000 - 90,000; Full time

Job summary

Inherentlabs is looking for a Member of Technical Staff to help build the training and inference systems necessary for cutting-edge AI research. This role involves developing core infrastructure for model training, emphasizing speed and reliability.

The successful candidate will collaborate closely with the research team and have experience in high-performance distributed systems, particularly with Python and AI technologies. A creative, technical culture with minimal bureaucracy awaits at Inherentlabs.

Qualifications

• Experience with large-scale distributed systems for AI workloads.
• Strong proficiency in Python, PyTorch, or JAX.
• Knowledge of distributed job orchestration and data pipelines.

Responsibilities

• Contribute to training and inference infrastructure end-to-end.
• Build systems for large-scale experiments, including data pipelines.
• Implement RL and post-training libraries for models.

Skills

High-performance distributed systems experience

Proficiency in Python and PyTorch or JAX

Experience with large-scale LLM training/inference technologies

Good judgment on building, buying, or deleting tools

AI-focused mindset

Tools

Python

PyTorch

JAX

Rust

C++

CUDA

Job description

Inherentlabs is looking for a Member of Technical Staff to help build the training and inference systems necessary for cutting-edge AI research. This role involves developing core infrastructure for model training, emphasizing speed and reliability.

The successful candidate will collaborate closely with the research team and have experience in high-performance distributed systems, particularly with Python and AI technologies. A creative, technical culture with minimal bureaucracy awaits at Inherentlabs.