AI Research Engineer (PyTorch / Transformers / Rust)

Format: remote, full-time, long-term contract
Time zone: working hours overlapping with UK time (GMT+0) Company: fast-growing, well-funded product company (developer tools / data platform)

Rate: $30–40/h (contractor, net)

About the role
The company is building a "self-driving" product and is training its own deep learning models on petabytes of product data: events, errors, session replays and more. You will join a new AI research team with real autonomy and direct influence on what gets shipped. This is a research-to-production role: shipping comes first.

What you will work on

• Session replay analysis: faster, cheaper models for detecting user issues, trained on large-scale DOM mutation data

• User behavior prediction: spotting drop-offs and suggesting improvements automatically

• Synthetic user testing: simulating users to catch confusing flows or breakages before release

• Training, evaluating and optimizing transformer-based models, from experiments to production

Requirements

• Strong math background (PhD is a plus) or hands-on experience training models
• Strong PyTorch

• Strong coding skills in Rust, CUDA or C (Rust preferred)

• Solid understanding of Transformer architectures

• Product mindset: you build things that ship, not just notebooks
• Fluent English

Not a fit

• Purely theoretical profiles
• Research-only focus without interest in shipping to production