Machine Learning Engineer – Pre Training

Job not on LinkedIn

🕒 May 26

Apply Now
Find Similar Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Logo of Mindbeam AI

Mindbeam AI

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.

📋 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

Similar Jobs

🕒 May 25

Affirm

1001 - 5000

💳 Fintech

👥 B2C

🛍️ eCommerce

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.

🕒 May 20

AssetWorks Inc

5001 - 10000

☁️ SaaS

🏢 Enterprise

🚗 Transport

Senior Full-Stack Developer leading the technical realization of generative AI roadmap for asset management solutions. Building tools that enhance expert judgment through sophisticated Agentic workflows.

🕒 May 20

AssetWorks Inc

5001 - 10000

☁️ SaaS

🏢 Enterprise

🚗 Transport

AI/ML Engineer leading technical implementation of generative AI systems for enterprise, designing multi-agent systems and overseeing AI workflows, ensuring AI implementations are scalable and secure.

🕒 May 17

Paramount

10,000+ employees

📱 Media

👥 B2C

Machine Learning Engineer optimizing onboarding and re-entry experiences within Paramount's streaming services. Focusing on user commitment and personalization through advanced machine learning techniques.

🕒 May 15

Camus Energy

11 - 50

⚡ Energy

🤖 Artificial Intelligence

☁️ SaaS

Machine Learning Engineer advancing forecasting and predictive modeling in clean energy software platform. Owning the full lifecycle of ML model development with strong product influence.