Software Engineer MLX

webAI; United States; Remote; USD 100,000 - 140,000; Full time

Job summary

A leading company in AI is seeking a Software Engineer MLX to optimize machine learning models for iOS deployment. The role involves using C++, Python, and Apple's MLX and Metal frameworks to enhance mobile AI solutions. Candidates should have a strong background in machine learning and hardware-aware programming. The company values truth, ownership, tenacity, and humility, and offers competitive benefits.

Benefits

Competitive salary and performance-based incentives

Comprehensive health, dental, and vision benefits package

401k Match

$200/mos Health and Wellness Stipend

$400/year Continuing Education Credit

Free parking, for in-office employees

Unlimited Approved PTO

Supplemental Life Insurance

Qualifications

• Experience with SIMD vectorization and multi-threading.
• Strong background in computer architecture.
• Hands-on experience in Machine Learning and Deep Learning.

Responsibilities

• Optimize and deploy machine learning models on iOS devices.
• Develop high-performance, hardware-aware code in C++ and/or Python.
• Collaborate with AI researchers and mobile teams.

Skills

C++

Python

Machine Learning

Deep Learning

Data Science

Hardware-aware programming

Education

Bachelor’s, Master’s, or Ph.D. in Computer Science

Electrical/Electronics/Telecommunications Engineering

Information Technology

Tools

MLX

Metal

PyTorch

Job description

webAI is pioneering the future of artificial intelligence by establishing the first distributed AI infrastructure dedicated to personalized AI. We recognize the evolving demands of a data-driven society for scalability and flexibility, and we firmly believe that the future of AI lies in distributed processing at the edge, bringing computation closer to the source of data generation. Our mission is to build a future where a company's valuable data and intellectual property remain entirely private, enabling the deployment of large-scale AI models directly on standard consumer hardware without compromising the information embedded within those models. We are developing an end-to-end platform that is secure, scalable, and fully under the control of our users, empowering enterprises with AI that understands their unique business. We are a team driven by truth, ownership, tenacity, and humility, and we seek individuals who resonate with these core values and are passionate about shaping the next generation of AI.

About the Role:

WebAI is hiring a Software Engineer MLX to optimize machine learning models for iOS deployment. You’ll use your expertise in C++, Python, hardware-aware programming, and Apple’s MLX and Metal frameworks to accelerate performance on mobile devices. If you’re passionate about building efficient, on-device AI solutions, we want to hear from you.

Key Responsibilities:

• Optimize and deploy machine learning models on iOS devices using MLX and Metal frameworks.
• Develop high-performance, hardware-aware code in C++ and/or Python, focusing on vectorization, multi-threading, and system optimization.
• Build and optimize custom Metal kernels and computational graphs for mobile acceleration.
• Deploy and fine-tune PyTorch models for efficient, on-device inference.
• Apply quantization, tensor fusion, and batching strategies to maximize model performance and minimize size.
• Collaborate with AI researchers and mobile teams to deliver scalable, production-ready solutions.
• Monitor and improve model performance across the iOS lifecycle to ensure efficiency and reliability.

Required Skills & Qualifications:

• Bachelor’s, Master’s, or Ph.D. in Computer Science, Information Science,
• Electrical/Electronics/Telecommunications Engineering, Information Technology, or a related field — or equivalent work experience.
• Strong background in computer architecture and hardware-aware programming.
• Proficiency in C++ and/or Python, with a focus on optimizing computational workloads.
• Experience with SIMD vectorization, multi-threading, and low-level performance tuning.
• Hands-on experience in Machine Learning, Deep Learning, and/or Data Science.

Preferred Qualifications:

• Familiarity with the iOS application lifecycle and memory optimization.
• Experience deploying PyTorch models on mobile devices, specifically iOS.
• Proficiency with Apple’s MLX framework for machine learning acceleration.
• Deep knowledge of Metal and Metal Performance Shaders (MPS).
• Experience developing custom Metal kernels for specialized operations.
• Strong understanding of computational graph optimization, batching strategies, device training, and fine-tuning.
• Knowledge of quantization techniques, tensor operation fusion, and graph reconstruction to optimize model performance and size.

We at webAI are committed to living out thecore values we have put in place as the foundation on which we operate as a team. We seek individuals who exemplify the following:

• Truth - Emphasizing transparency and honesty in every interaction and decision.
• Ownership - Taking full responsibility for one’s actions and decisions, demonstrating commitment to the success of our clients.
• Tenacity - Persisting in the face of challenges and setbacks, continually striving for excellence and improvement.
• Humility - Maintaining a respectful and learning-oriented mindset, acknowledging the strengths and contributions of others.
• Competitive salary and performance-based incentives.
• Comprehensive health, dental, and vision benefits package.
• 401k Match
• $200/mos Health and Wellness Stipend
• $400/year Continuing Education Credit
• Free parking, for in-office employees
• Unlimited Approved PTO
• Supplemental Life Insurance

webAI is an Equal Opportunity Employer and does not discriminate against any employee or applicant on the basis of age, ancestry, color, family or medical care leave, gender identity or expression, genetic information, marital status, medical condition, national origin, physical or mental disability, protected veteran status, race, religion, sex (including pregnancy), sexual orientation, or any other characteristic protected by applicable laws, regulations and ordinances. We adhere to these principles in all aspects of employment, including recruitment, hiring, training, compensation, promotion, benefits, social and recreational programs, and discipline. In addition, it is the policy of webAI to provide reasonable accommodation to qualified employees who have protected disabilities to the extent required by applicable laws, regulations and ordinances where a particular employee works.

Create a Job Alert

Interested in building your career at webAI? Get future opportunities sent straight to your email.

Accepted file types: pdf, doc, docx, txt, rtf

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Education

School * Select...

Degree * Select...

Select...

Start date year

End date month Select...

End date year

Are you legally authorized to work in the United States?Note: If hired, a Form I-9, Employment Eligibility Verification, must be completed at the start of employment. * Select...

Do you now, or will you in the future require sponsorship for employment (e.g;- H-1 B visa, Work Permit, etc) in the United States? * Select...

Do you have strong proficiency in C++ software development? * Select...

Do you have 5+ years of experience in Software Engineering with a focus on embedded systems or AI/ML. *

Do you have professional experience with MLX? *

If you had to pick one project to highlight your recent experience, what would it be? (please provide a 2-4 sentence description) *