Applied AI Engineer

About the Company

A1 is on a mission to build a proactive smart assistant for everyday users, bringing intelligence to conversations, errands, organizing, and workflows with minimal prompting. The product focuses on high reliability for long-running workflows, persistent context, and real-world task completion.

Responsibilities

• Build and ship AI features end-to-end from model to user experience
• Design and iterate on prompts, tools, memory, and agent workflows
• Turn raw model outputs into structured, reliable, and predictable behaviors
• Debug issues across the full stack including model, orchestration, infra, and UX
• Optimize for latency, cost, and production reliability
• Develop lightweight evaluation frameworks to measure real-world performance
• Work closely with product and engineering to translate ambiguous problems into working systems

Requirements

• Strong foundation in machine learning and modern neural network architectures
• Hands-on experience with training, fine-tuning, or deploying ML models
• Ability to write clean, production-quality code
• Comfort working across abstraction layers from model to infra to product
• Strong problem-solving skills in ambiguous, fast-moving environments
• Bias toward shipping, iteration, and continuous improvement

Preferred Qualifications

• Proficiency with Python
• Experience with PyTorch or JAX
• Experience with LLMs (OpenAI-style APIs, LLaMA, Qwen, etc.)
• Experience with inference/serving tools like vLLM
• Experience with Vector Databases