Machine Learning Engineer Jobs with Visa Sponsorship

Anthropic

As a TPU Kernel Engineer, you will be responsible for identifying and addressing performance issues across various ML systems, including research, training, and inference. A significant part of this role involves designing and optimizing kernels specifically for the TPU. You will also provide crucial feedback to researchers regarding how model changes affect performance. The ideal candidate will have a proven history of solving large-scale systems problems and demonstrating expertise in low-level optimization.

$280k - $850k

Thinking Machines Lab

We are seeking a Software Engineer, Inference to be responsible for the reliability, scale, and efficiency of the systems that serve our AI models to users. This production-facing systems role is central to the company, acting as a bridge between cutting-edge inference techniques and the practicalities of serving live traffic, including rollouts, capacity management, and incident response. You will ensure that performance gains from our research and inference teams reach production safely and remain operational, powering our live serving platform and related products.

$300k - $400k

Anthropic

The Account Abuse team is responsible for ensuring Anthropic's computing capacity is allocated fairly by minimizing resources for bad actors and preventing their return. As a software engineer on this team, you will build the machine learning systems that detect and stop abuse at scale. The ideal candidate can anticipate opponent strategies and responses to countermeasures. This role involves full-stack machine learning engineering across model training, productionization, and evaluation, with opportunities to leverage Claude to accelerate model development and maintenance. The focus is on classical ML applied to structured and behavioral data, emphasizing robust production systems, precise measurement, and safe rollout strategies, as false positives can lead to legitimate customer lockout.

$320k - $485k

Anthropic

The Safeguards ML Inference Path team designs, builds, and operates the production infrastructure that powers Claude's ML-based safety systems. This role involves collaborating closely with safety researchers and inference engineers to bring new classifiers and novel classes of ML defenses to production, owning the research-to-production transfer of new safety technologies critical for every Claude model launch. The team builds for scale, serving thousands of ML classifiers for all requests on the token generation path and for every platform Claude runs on. Engineers will work at the intersection of machine learning, large-scale distributed systems, and AI safety, developing platforms and tools that enable safeguards to operate reliably at scale, with tooling and infrastructure used for every model launch.

$320k - $485k

Nunchux AI is building infrastructure to make multimodal generative AI faster, cheaper, and easier to build on. Founded by MIT PhDs and CMU Professor, the company leverages deep research expertise and production systems experience. The Role focuses on bringing new image and video models to the Modelverse platform, prioritizing speed to launch. You will own the initial integration, coordinate deployment with the cloud team, and work with product engineers on API and Modelverse integration. After launch, you will hand off optimization tasks and maintain the shared code for model integration and serving.

$170k - $240k

Nunchux AI is building infrastructure to make multimodal generative AI faster, cheaper, and easier to build upon. We are seeking a Machine Learning Engineer focused on post-training and evaluation to enhance the efficiency and quality of visual generative models. In this role, you will develop and validate post-training techniques, curate and manage data pipelines for training and evaluation, and build robust evaluation systems to measure model performance. You will also benchmark models against industry standards, ensure quality, and integrate cutting-edge research into production workflows.

$180k - $250k