AI ML Specialist Lead

We are seeking a cross-functional expert at the intersection of Product, Engineering, and Machine Learning to lead and build cutting-edge AI systems. This role combines the strategic vision of a Product Manager with the technical expertise of a Machine Learning Engineer and the innovation mindset of a Generative AI and LLM expert.

You will help define, design, and deploy AI-powered features, train and fine-tune models (including LLMs), and architect intelligent AI agents that solve real-world problems at scale.

Key Responsibilities:

Product Management:

Define product vision, roadmap, and AI use cases aligned with business goals.

Collaborate with cross-functional teams (engineering, research, design, business) to deliver AI-driven features.

Translate ambiguous problem statements into clear, prioritized product requirements.

⚙️ AI/ML Engineering & Model Development:

Develop, fine-tune, and optimize ML models, including LLMs (GPT, Claude, Mistral, etc.).

Build pipelines for data preprocessing, model training, evaluation, and deployment.

Implement scalable ML solutions using frameworks like PyTorch, TensorFlow, Hugging Face, LangChain, etc.

Contribute to R&D for cutting-edge models in GenAI (text, vision, code, multimodal).

AI Agents & LLM Tooling:

Design and implement autonomous or semi-autonomous AI Agents using tools like AutoGen, LangGraph, CrewAI, etc.

Integrate external APIs, vector databases (e.g., Pinecone, Weaviate, ChromaDB), and retrieval-augmented generation (RAG).

Continuously monitor, test, and improve LLM behavior, safety, and output quality.

Data Science & Analytics:

Explore and analyze large datasets to generate insights and inform model development.

Conduct A/B testing, model evaluation (e.g., F1, BLEU, perplexity), and error analysis.

Work with structured, unstructured, and multimodal data (text, audio, image, etc.).

Preferred Tech Stack / Tools:

Languages: Python, SQL, optionally Rust or TypeScript

Frameworks: PyTorch, Hugging Face Transformers, LangChain, Ray, FastAPI

Platforms: AWS, Azure, GCP, Vertex AI, Sagemaker

ML Ops: MLflow, Weights & Biases, DVC, Kubeflow

Pandas, NumPy, Spark, Airflow, Databricks

Vector DBs: Pinecone, Weaviate, FAISS

Model APIs: OpenAI, Anthropic, Google Gemini, Cohere, Mistral

Tools: Git, Docker, Kubernetes, REST, GraphQL

‍ Qualifications:

Bachelor’s, Master’s, or PhD in Computer Science, Data Science, Machine Learning, or a related field.

10+ years of experience in core ML, AI, or Data Science roles.

Proven experience building and shipping AI/ML products.

Deep understanding of LLM architectures, transformers, embeddings, prompt engineering, and evaluation.

Strong product thinking and ability to work closely with both technical and non-technical stakeholders.

Familiarity with GenAI safety, explainability, hallucination reduction, and prompt testing, computer vision

Bonus Skills:

Experience with autonomous agents and multi-agent orchestration.

Open-source contributions to ML/AI projects.

Prior startup or high-growth tech company experience.

Knowledge of reinforcement learning, diffusion models, or multimodal AI.