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.

Airflow

Python

🕒 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.

Azure

Cloud

Microservices

Postgres

Python

SQL

.NET

🕒 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.

Azure

Cloud

Microservices

Postgres

Python

SQL

.NET

🕒 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.

Google Cloud Platform

PyTorch

Tensorflow

🕒 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.

Pandas

Python

PyTorch

Scikit-Learn