Software Engineer, webAI
MLX
Unified execution layer for AI
Be an early applicant
Unified execution layer for AI
21-100 employees
Be an early applicant
21-100 employees
Company mission
To revolutionize the AI industry by laying the foundation for the development of Super Intelligence (SI).
Role
Who you are
- 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
Desirable
- 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
What the job involves
- 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
- 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
Our take
On a mission to democratize artificial intelligence, WebAI has built a decentralised development platform that minimises the amount of data required to build AI models. This directly challenges the assumption that big data is intrinsic to AI, and enables for local, ownable and customizable models to be made.
The company's unified execution layer helps to streamline the training and deployment of models in a way which is both scalable and flexible. Enterprises can use their existing infrastructure to run powerful models while ensuring their data never leaves their devices. Plus they can tailor each model to their needs with no coding needed.
WebAI believes that off-the-shelf AI models lack the flexibility and control required to be used in real-world tasks, which is what its platform helps to combat. This approach has attracted impressive investment that will be used to grow its development efforts and elevate its mission to democratise AI.
Steph
Company Specialist at Welcome to the Jungle
Company
Funding (2 rounds)
Sep 2024
$60m
SERIES A
Dec 2022
$17m
SEED
Total funding: $77m
Company benefits
- Performance-based incentives
- Comprehensive health, dental, and vision benefits package
- $200/mos Health and Wellness Stipend
- $400/year Continuing Education Credit
- Flexible work week
- Free parking for in-office employees
- Unlimited PTO
- Parental & Bereavement Leave
- Supplemental Life Insurance
Company HQ
Congress Avenue Historic District, Austin, TX
Leadership
David Stout
(Co-Founder & CEO)Also Founder at Ghost House Technology.
Ethan Baird
(Co-Founder & Chief Research Officer)Previously Software Programmer at Ghost House Technology and Web Technician at Revision Marketing and Design.
Tyler Mauer
(Co-Founder & Chief Innovation Officer)Previously Director of Product and Market at Ghost House Technology, and consulted at YETi, RANDY'S HOUSE and Shinola.
Share this job
View 18 more jobs at webAI