ML Infrastructure Engineer

45 minutes ago
Full-time
Junior
DevOps and Infrastructure
x.ai

x.ai

x.ai is a technology company founded in 2014 that specializes in developing artificial intelligence personal assistants. Their flagship product is a personal assistant that efficiently schedules meetings for users. With a focus on sustainable business ...

Internet Software & Services
51-250
Founded 2023
$44M raised

Description

  • Design, build, and scale GPU compute infrastructure, training frameworks, and experimentation tools.
  • Develop data pipelines and integrate large-scale data, training, and inference systems.
  • Collaborate with ML teams to productionize models and integrate them across the stack.
  • Ensure scalability, reliability, and efficiency of large-scale machine learning systems.
  • Work across the full stack to solve complex problems independently.
  • Mentor junior engineers and help grow the team.

Requirements

  • Bachelor’s, Master’s, post-graduate, or PhD in computer science, machine learning, or another quantitative discipline, or equivalent work experience.
  • 2+ years of industry experience in high-traffic or large-scale production environments, distributed systems, GPU infrastructure, and/or deep learning applications.
  • 2+ years of experience with ML platforms, training infrastructure, or close collaboration with modeling engineers and data scientists.
  • Strong proficiency with Python.
  • Experience with compiled languages such as C++ or Rust.
  • Deep familiarity with modern ML frameworks such as JAX or PyTorch (preferred).
  • Low-level understanding of compute systems, including distributed storage, NVIDIA drivers, CUDA toolkits, and networking (preferred).
  • Comfort with Linux systems and orchestration tools (preferred).
  • Experience with job schedulers such as Slurm or configuration management tools such as Puppet/Ansible (preferred).

Benefits

  • $180,000 - $440,000 USD base salary.
  • Equity.
  • Comprehensive medical, vision, and dental coverage.
  • 401(k) retirement plan.
  • Short- and long-term disability insurance.
  • Life insurance.
  • Various additional discounts and perks.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Machine Learning Engineer, Safety

Reddit 1K-5K Internet Software & Services

Reddit is hiring a remote-friendly Machine Learning Engineer to build and improve safety systems that support enforcement of Reddit rules using large language models.

Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
1 hour, 15 minutes ago

Staff Network Engineer (AI Fabric, Datacenter and Edge Networking) - Radian Arc (EMEA)

Submer 51-250 IT Services

Radian Arc is hiring a Staff Network Engineer to design and operate the networking infrastructure behind its GPU cloud platform for AI, machine learning, and cloud gaming workloads in telecom carrier networks.

Bash Python PyTorch TensorFlow WAF
1 hour, 30 minutes ago

Cloud Platform Architect (AI Infrastructure Environments)

Gramian Consultancy Group Professional Services

Gramian Consultancy is hiring a fully remote Cloud Platform Architect to build realistic cloud engineering environments used to evaluate how advanced AI systems solve production infrastructure problems.

AWS Azure GCP Kubernetes Terraform
1 hour, 30 minutes ago

Senior, Machine Learning Engineer - 3D Perception

Torc 251-1K Road & Rail

Torc is hiring a Senior Machine Learning Engineer – 3D Perception to develop and deploy production perception models for autonomous trucks and improve Bird's Eye View understanding across its autonomy stack.

C++ Computer Vision Deep Learning Machine Learning Python PyTorch
1 hour, 30 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers