Applied AI Engineer

About this role

A1 is seeking an Applied AI Engineer to build production-ready AI features for a smart assistant platform. You'll work across the full stack—from model optimization to system design to user experience—ensuring AI workflows are reliable, fast, and actually work in the real world.

What you'll do

• Design and iterate on prompts, tools, memory, and agent workflows to shape model behavior
• Build end-to-end AI features from model selection through production deployment
• Debug issues across model, orchestration, infrastructure, and UX layers
• Create lightweight evaluation frameworks to measure real-world performance
• Optimize for latency, cost, and production reliability
• Collaborate with product and engineering to translate ambiguous problems into working systems

What they're looking for

• Python and production-quality code
• PyTorch or JAX
• LLM APIs and model fine-tuning experience
• ML inference and serving (e.g., vLLM)
• Vector databases
• Machine learning fundamentals and neural networks
• Full-stack debugging across abstraction layers
• Problem-solving in ambiguous, fast-moving environments