Taro Logo

Software Engineer MLX

WebAI is pioneering distributed AI infrastructure for personalized AI, focusing on edge processing and private data deployment.
Mobile
Mid-Level Software Engineer
Remote
3+ years of experience
AI

Description For Software Engineer MLX

WebAI is at the forefront of distributed AI infrastructure development, focusing on personalized AI solutions that prioritize data privacy and edge computing. As a Software Engineer MLX, you'll be instrumental in optimizing machine learning models for iOS deployment, working with cutting-edge technologies like Apple's MLX and Metal frameworks. The role combines expertise in C++, Python, and hardware-aware programming to create efficient, on-device AI solutions. You'll be part of a team driven by core values of truth, ownership, tenacity, and humility, working on challenging problems in AI model optimization and deployment. The position offers comprehensive benefits including competitive compensation, health benefits, professional development opportunities, and flexible PTO. This is an excellent opportunity for engineers passionate about mobile AI optimization and interested in shaping the future of distributed artificial intelligence.

Last updated a day ago

Responsibilities For Software Engineer MLX

  • 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
  • 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
  • Collaborate with AI researchers and mobile teams
  • Monitor and improve model performance across the iOS lifecycle

Requirements For Software Engineer MLX

Python
TypeScript
  • Bachelor's, Master's, or Ph.D. in Computer Science or related field
  • Strong background in computer architecture and hardware-aware programming
  • Proficiency in C++ and/or Python
  • Experience with SIMD vectorization, multi-threading, and low-level performance tuning
  • Hands-on experience in Machine Learning, Deep Learning, and/or Data Science

Benefits For Software Engineer MLX

401k
Dental Insurance
Medical Insurance
Vision Insurance
Education Budget
Parental Leave
  • Competitive salary and performance-based incentives
  • Comprehensive health, dental, and vision benefits package
  • 401k Match
  • $200/month Health and Wellness Stipend
  • $400/year Continuing Education Credit
  • Free parking for in-office employees
  • Unlimited Approved PTO
  • Parental, Bereavement Leave
  • Supplemental Life Insurance

Interested in this job?

Jobs Related To webAI Software Engineer MLX