Senior AI Software Engineer

About Us:

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:

We are looking for a Senior AI Software Engineer with deep expertise in high-performance computing and machine learning inference. In this role, you’ll design and optimize inference engines for on-device applications on Apple platforms (macOS & iOS), leveraging Metal and MLX to push performance to its limits.

This is an ideal opportunity for someone who thrives at the intersection of low-level systems programming, applied machine learning, and hardware-aware optimization.

Key Responsibilities

- Implement and optimize advanced ML architectures (Transformers, Mixture of Experts, Diffusion models) in C++, with a focus on performance and memory efficiency.

- Develop and fine-tune custom Metal kernels for performance-critical inference operations.

- Apply advanced model quantization techniques (low-bit, mixed-precision) to accelerate performance while minimizing footprint.

- Profile, benchmark, and tune inference on Apple Sili...