Senior ML Engineer

Hiring via TechTree

This is a role that TechTree is recruiting for on behalf of one of its clients.

TechTree is an AI-driven recruitment platform working with high-growth companies.

When you apply TechTrees AI Agent matches you not just to this role but to other relevant opportunities across its network so one application can unlock multiple roles.

Senior ML Engineer Foundation Models & CUDA

TechTrees client is hiring a Senior ML Engineer to help build and scale a novel foundation model for automated software delivery in embedded systems.

This is a deeply technical role for an engineer who has built large-scale foundation models developed custom CUDA kernels and scaled distributed ML systems in production.

• Location: Brazil remote
• Employment: Full-time
• Estimated compensation: /year equity
• Level: Mid-Senior / Senior
• Relocation: UK visa sponsorship may be available

What youll own

• Lead development and production deployment of a large-scale foundation model
• Design and implement custom CUDA kernels for performance-critical workloads
• Optimise training and inference across GPUs and distributed infrastructure
• Architect scalable ML systems with demanding reliability and performance requirements
• Profile and improve data pipelines training loops inference and serving infrastructure
• Evaluate modern architectures including Mixture-of-Experts and state-space models
• Build internal tooling benchmarks evaluation systems and observability infrastructure
• Work closely with founders on technical strategy and optimisation priorities
• Improve model quality latency scalability and compute efficiency

What were looking for

• Deep professional experience with CUDA C/C
• Strong Python skills
• Experience building and shipping large-scale foundation models
• Proven custom CUDA kernel development
• Experience with distributed training and inference
• Strong knowledge of modern deep learning architectures
• Production expertise with PyTorch or another major ML framework
• Experience operating ML workloads on AWS Azure or GCP
• Strong understanding of ML observability evaluation and production SLOs
• Experience delivering complex systems in fast-moving environments

Strong advantages

• Built a foundation model from 01
• Early-stage AI startup or top-tier research lab experience
• Experience with MoE state-space models and advanced inference optimisation
• Built tooling that significantly improved engineering or research productivity
• Experience optimising large-scale GPU workloads for cost and performance

Why consider this opportunity

• Work on a genuinely novel foundation-model problem
• Significant technical ownership from day one
• Direct influence over architecture and ML strategy
• Work closely with an experienced founding team
• Competitive compensation plus equity
• Remote opportunity for candidates based in Brazil
• Potential UK relocation support

If this sounds like a fit wed love to hear from you. Apply today!

Employment Type : Full Time
Experience: years
Vacancy: 1
Yearly Salary Salary: 100000 - 150000