Lead AI Engineer

You might be our missing piece if you have:

• Deep expertise in Python and AI frameworks such as PyTorch (ideal), TensorFlow, Keras, or SciPy.

• Production-grade experience with FastAPI, Flask, or Django for serving AI models and building API-driven applications.

• Proven experience in architecting and scaling ML systems in production—everything from data pipelines and model deployment to monitoring and optimization.

• Experience with AI/ML architectures and AI-powered product architectures including: multi-agent architectures, multi-headed models, inference architectures. 

• Experience with connecting data pipelines and serving inference processes.

• Strong hands-on knowledge of Docker and an understanding of containerized ML workflows (DevOps builds it, but you speak its language).

• Proficiency in SQLAlchemy, Alembic, and relational databases, along with a solid understanding of how to model, move, and optimize data for ML use cases.

• Expertise in data wrangling and preprocessing using tools like Pandas, NumPy, or SciPy.

• Experience integrating with and building on external AI APIs (e.g., OpenAI, Anthropic, Hugging Face, etc.).

• A solid understanding of software engineering principles, versioning (Git), and Python code style (PEP 8).

• A track record of leading AI/ML projects from ideation to deployment.

• Experience mentoring and guiding engineers, helping them grow technically and strategically.

We would be thrilled if you have:

• Exposure to research or academic papers, and a knack for translating theory into production-ready code.

• Comfort with cloud infrastructure (AWS, Azure, GCP, or DigitalOcean) and understanding of CI/CD best practices.

• Hands-on experience with LLMs, embeddings, semantic search, and tools like StableDiffusion or LLaMA.

• A collaborative mindset and the ability to bridge the gap between product, engineering, and research teams.

We will be working together on the following:

• Understanding clients' and their requirements and providing optimal solutions.

• Staying ahead of industry trends and implementing cutting-edge solutions.

• Technical leadership over major AI initiatives, from architecture design to final delivery.

• Mentoring and upskilling the AI team—pair programming, code reviews, and continuous learning are your bread and butter.

• Driving best practices around model development, reproducibility, monitoring, and scalability.

• Staying up to date with state-of-the-art advancements and continuously leveling up our AI capabilities.