Machine Learning Engineer - Pre Training

Posted 2 Days Ago
Hiring Remotely in United States
Remote
150K-190K Annually
Junior
Artificial Intelligence • Machine Learning • Software • Generative AI
The Role
Build and optimize scalable pre-training pipelines for foundation models. Implement distributed training across GPUs, TPUs, and high-performance clusters; translate research experiments into production workflows; develop monitoring and fault-tolerance systems; and benchmark and tune performance across hardware and software stacks.
Summary Generated by Built In

About Mindbeam

We are building the next-generation AI infrastructure for open source and enterprise. Our work is deeply research-oriented and passionate about developing ground-breaking innovations to take state-of-the-art AI applications to the next level.

What drives us is not only advancing technology, but empowering the people behind it. We are a community of researchers, engineers, and visionaries who believe that collaboration, curiosity, and openness fuel progress. If you’re motivated by impact and inspired to build tools that others can build upon, you’ll be in the right place.

Mission

Design and optimize large-scale pre-training systems that power Mindbeam’s generative AI models.

Role Expectations

• Build scalable pre-training pipelines for foundation models, optimizing throughput and efficiency.

• Implement distributed training strategies across GPUs/TPUs and high-performance clusters.

• Collaborate with researchers to translate experimental setups into production-ready workflows.

• Develop monitoring and fault-tolerance systems to ensure reliable large-scale training.

• Continuously benchmark and tune performance across hardware and software stacks.

Background

• Bachelor’s, Master’s, or PhD in Computer Science, Engineering, or related field—or equivalent experience.

• 2+ years of experience with large-scale model training and distributed systems.

• Strong coding skills in Python and familiarity with ML frameworks (PyTorch, TensorFlow, JAX).

• Experience with GPU scheduling, memory optimization, and parallelism strategies.

• Comfort with containerized and orchestrated environments (Docker/Kubernetes).

• Understanding of high-performance computing and networking bottlenecks.

About You

You thrive on scale and complexity. You enjoy solving system-level bottlenecks, pushing hardware and software to their limits, and working closely with researchers to accelerate cutting-edge AI development.

Skills Required

  • Bachelor's, Master's, or PhD in Computer Science, Engineering, or a related field, or equivalent experience
  • 2+ years of experience with large-scale model training and distributed systems
  • Strong coding skills in Python
  • Familiarity with PyTorch, TensorFlow, and JAX
  • Experience with GPU scheduling, memory optimization, and parallelism strategies
  • Experience with containerized and orchestrated environments using Docker and Kubernetes
  • Understanding of high-performance computing and networking bottlenecks
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
11 Employees
Year Founded: 2024

What We Do

Mindbeam AI builds next-generation AI infrastructure for generative AI applications. Its Litespark framework uses advanced algorithms to accelerate language-model training and inference workloads while remaining compatible with standard machine-learning frameworks. The company also develops high-throughput, low-latency inference technology designed to reduce memory usage and support deployment across hardware such as consumer CPUs. Its work focuses on making AI development faster and more efficient.

Similar Jobs

Remote
USA
62 Employees
90-105 Hourly
In-Office or Remote
Chicago, IL, USA
1934 Employees
133K-170K Annually

Vannevar Logo Vannevar

Senior Software Engineer

Artificial Intelligence • Machine Learning • Software • Defense
Remote
USA
225 Employees

Citizens Logo Citizens

Wealth Advisor - Johnston, RI

Digital Media • Fintech • Information Technology • Machine Learning • Financial Services • Cybersecurity • Automation
In-Office or Remote
2 Locations
17000 Employees
105K-250K Annually

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
70 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account