Machine Learning Engineer – Pre Training

Mindbeam AI

Design and optimize large-scale pre-training systems powering generative AI models at Mindbeam. Collaborate with researchers and build efficient training pipelines.

Posted 5/26/2026full-timeRemote • 🇺🇸 United StatesJuniorMid-Level💰 $150,000 - $190,000 per yearWebsite

Tech Stack

Tools & technologies

Distributed SystemsDockerKubernetesPythonPyTorchTensorflow

About the role

Key responsibilities & impact

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.

Requirements

What you’ll need

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.

Benefits

Comp & perks

🌐 Worldwide ❌ Jobs You've Hidden ⭐️ Saved Jobs ✅ Applied Jobs ✉️ Email Alerts 👤 Account Mindbeam AI Website LinkedIn All Job Openings 1 - 10 employees Founded 2024 🤖 Artificial Intelligence 🤝 B2B Artificial Intelligence
B2B Mindbeam AI is a company building Litespark, an ultra-fast LLM pretraining framework that accelerates training and inference for generative AI applications. Its software improves throughput on existing GPU hardware with zero code changes, is compatible with industry-standard ML frameworks like PyTorch, and claims to reduce pretraining time from months to days while lowering energy consumption and costs (up to ~81% energy savings in cited workloads). Mindbeam targets enterprise and B2B customers seeking scalable, efficient AI infrastructure for large language model development. Machine Learning Engineer – Pre Training Job not on LinkedIn 🔥 30 minutes ago 🇺🇸 United States – Remote 💵 $150k - $190k / year ⏰ Full Time 🟢 Junior 🟡 Mid-level 🤖 Machine Learning Engineer Distributed Systems Docker Kubernetes Python PyTorch Tensorflow Apply Now Find Hiring Managers Customize resume + cover letter Report problem ☆ Save ☑️ Mark as applied ❌ Hide 📋 Description
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. 🎯 Requirements
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. Apply Now 📊 Check your resume score for this job Improve your chances of getting an interview by checking your resume score before you apply. Check Resume Score Similar Jobs Machine Learning Engineer II 🕒 Yesterday Affirm 1001 - 5000 💳 Fintech 👥 B2C 🛍️ eCommerce Website LinkedIn All Job Openings Machine Learning Engineer optimizing and automating financial customer operations at Affirm. Collaborating across teams to develop and deploy AI systems for dispute and chargeback handling. 🇺🇸 United States – Remote 💵 $142k - $210k / year 💰 Post-IPO Equity on 2021-01 ⏰ Full Time 🟢 Junior 🟡 Mid-level 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Airflow Python MLOps, Catastrophe and Geo-Analytics 🕒 4 days ago Mercury Insurance 5001 - 10000 💸 Finance 👥 B2C Website LinkedIn All Job Openings MLOps Engineer developing machine learning applications for Mercury Insurance. Responsible for productionizing models, handling data science pipelines, and collaborating with data teams. 🇺🇸 United States – Remote 💵 $94.5k - $179k / year ⏰ Full Time 🟢 Junior 🟡 Mid-level 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor AWS Cloud Python SQL Manager, Forward Deployed Machine Learning Engineering 🕒 5 days ago Federato 11 - 50 🤖 Artificial Intelligence Website LinkedIn All Job Openings Manager of Forward Deployed Machine Learning Engineering leading AI and ML team delivering customer solutions. Ensuring operational excellence while collaborating across leadership in a fast-paced environment. 🇺🇸 United States – Remote 💵 $220k - $250k / year 💰 $15M Series A on 2022-09 ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Machine Learning Engineer 🕒 6 days ago Revelation Pharma 201 - 500 💊 Pharmaceuticals ⚕️ Healthcare Insurance 🤝 B2B Website LinkedIn All Job Openings Machine Learning Engineer designing, training, and monitoring ML models for healthcare applications. Owning model lifecycle management and collaborating with clinical teams for explainable models. 🇺🇸 United States – Remote 💵 $120k - $145k / year ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer AWS MLOps Engineer 🕒 6 days ago Guidehouse 10,000+ employees Website LinkedIn All Job Openings MLOps Engineer responsible for designing and implementing platforms for the deployment of machine learning solutions. Collaborating with data scientists and engineers to operationalize models in federal environments. 🇺🇸 United States – Remote 💵 113k - 188k / year 💰 Grant on 2023-02 ⏰ Full Time 🟡 Mid-level 🟠 Senior 🤖 Machine Learning Engineer 🦅 H1B Visa Sponsor Cloud Docker Kubernetes Python View More Machine Learning Engineer Jobs 🌐 Worldwide Built by Lior Neu-ner. I'd love to hear your feedback — Get in touch via DM or support@remoterocketship.com Search Search Jobs by country Search jobs by city Search jobs by job title Search entry-level jobs Search junior-level jobs Search senior-level jobs Search jobs by tech stack Search jobs by contract type Search remote internships Search remote part-time jobs Remote jobs Anywhere in the World Companies Hiring Anywhere in the World Companies Hiring Sales People Anywhere in the World Companies Hiring Software Engineers Anywhere in the World Resources Advice Tips for finding remote jobs Interview questions and answers Resume examples Cover letter examples Post a job Affiliates Privacy policy Terms of service Job board SEO course AI Apply Copilot OpenClaw job finder Jobs by Country Remote jobs anywhere in the world (Worldwide remote jobs) Remote jobs United States Remote jobs Australia Remote jobs Brazil Remote jobs Canada Remote jobs France Remote jobs Ireland Remote jobs Germany Remote jobs Netherlands Remote jobs Spain Remote jobs UK Popular Jobs Remote data analyst jobs Remote customer support jobs Remote executive assistant jobs Remote marketing jobs Remote product designer jobs Remote product manager jobs Remote project manager jobs Remote recruiter jobs Remote sales jobs Remote software engineer jobs Jobs by Type Remote full-time jobs Remote part-time jobs Remote contract jobs Remote internship jobs Remote entry-level jobs Remote jobs with no experience required Remote junior jobs (1-3 years of experience) Digital nomad jobs Remote jobs with no degree required Freelance remote jobs Temporary remote jobs Remote jobs hiring now Stay at home mom jobs

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

PythonPyTorchTensorFlowJAXGPU schedulingmemory optimizationparallelism strategiescontainerizationorchestrationhigh-performance computing