About the Position

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.

Responsibilities

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.