Senior AI-ML Data Scientist

Senior AI-ML Data Scientist

Job Summary

We are seeking a Senior AI/ML Data Scientist to own model and agent behavior end to end - from problem framing and algorithm selection through fine-tuning, retrieval design, agentic orchestration, evaluation, and production serving.

This is a hands-on role for someone with genuine depth in machine learning and statistics who is equally comfortable designing an experiment, reading an attention implementation, and shipping the result behind a latency budget. We are particularly interested in candidates who think carefully about agent memory - what an agent should retain, in what form, and how retention is grounded in a governed data warehouse rather than an undifferentiated vector blob.

This role partners closely with the AI Data Engineer, who owns the warehouse, pipelines, and index infrastructure. The boundary:

they own the pipeline, the schema, and the guarantees; you own the algorithm, the prompt, and the evaluation.

Required Qualifications

5–10+ years in ML/AI engineering, data science, or related technical roles, with proven experience deploying models at scale in production (LLM, CV, NLP, or multimodal).

ML depth

substantive command of machine learning algorithms and neural network theory -optimization, regularization, attention mechanisms, tokenization, embeddings, and model internals.

Statistics

rigorous grounding in inference, experimental design, and data analysis.

Frameworks

PyTorch (primary), plus TensorFlow or JAX; the Hugging Face ecosystem (Transformers, Datasets, TRL).

Python

expert-level, production-grade. Strong SQL for analysis against a dimensional warehouse.

Agentic systems

production experience with LangChain/LangGraph or equivalent, and a well considered position on agent memory architecture.

Knowledge graphs

hands-on ontology design and graph-based reasoning.

Cloud

• expert-level deployment of AI workloads on AWS, Azure, or GCP, including GPU provisioning, cost optimization, containerization, and CI/CD.
• Experience with experiment tracking and model lifecycle tooling (MLflow, Weights & Biases).

Preferred Qualifications

• Direct experience implementing CoALA or a comparable cognitive architecture (SOAR, ACT R, or a documented in-house framework) in a shipped agent system.
• GPU acceleration internals: CUDA, TensorRT, cuBLAS.
• Production experience with vLLM, NVIDIA Triton, Ray Serve/Ray Train, DeepSpeed, or FSDP.
• Experience with AI security, governance, and compliance frameworks.
• Track record of contributing to open-source AI frameworks, or published research.
• Ability to lead technical discovery phases and client-facing AI workshops.
• Familiarity with lakehouse table formats (Iceberg, Delta Lake) sufficient to collaborate credibly with data engineering.