We are seeking an experienced Machine Learning Lead to helm our Machine Learning team. In this pivotal role, you will be the engineering architect behind Vulcan’s core AI capabilities. You will act as the nexus between Research, Platform, and Product. Your mission is to translate cutting-edge findings on GenAI threats into robust, production-ready machine learning models that power our GenAI Security Guardrails (Blue Team) and Automated Vulnerability Assessment (Red Team).
🏢 About AIFT
AIFT is at the forefront of technology to provide security in GenAI, with opportunities to work on groundbreaking projects. We offer a competitive salary and benefits package, commensurate with experience and performance.
🎯 The Role
In this pivotal role, you will be the engineering architect behind Vulcan’s core AI capabilities. You will act as the nexus between Research, Platform, and Product. Your mission is to translate cutting-edge findings on GenAI threats into robust, production-ready machine learning models that power our GenAI Security Guardrails (Blue Team) and Automated Vulnerability Assessment (Red Team).
✅ Key Responsibilities
Collaborate with the Security Research Team to operationalize new threat detection techniques. They identify the \"what\" (e.g., new prompt injection patterns); you determine the \"how\" (model architecture, training strategy).
Lead the fine-tuning of Language Models (e.g., using LoRA/PEFT) to optimize for our supported multi-lingual languages and specific security intents.
Prepare the system for Multimodal (Text + Image/Audio) capabilities. Evaluate and implement models to detect visual prompt injections and non-textual threats as the product evolves.
Take ownership of our existing ML pipelines. Focus on optimizing and scaling CI/CD/CT workflows to improve training efficiency and deployment velocity.
Implement and enforce rigorous Data Versioning strategies (e.g., DVC) to ensure complete reproducibility of model artifacts and datasets.
Maintain rigorous monitoring for model drift and performance, ensuring high reliability in a production security environment.
Work closely with the Platform Engineering Team to integrate ML models into the broader product architecture. Ensure seamless interaction between model inference services and the main platform logic.
Lead and mentor Machine Learning Engineers, fostering a culture of engineering rigor, code quality, and operational excellence.
Manage GPU resources and compute budgets effectively for both training and inference workloads.
Act as the technical voice of the ML team. You must effectively explain complex ML concepts (e.g., compute complexity, infrastructure costs) into clear, business-relevant insights for decision-makers.
Justify compute resource investments. Articulate the trade-off between infrastructure costs (GPU hours) and performance gains to non-technical stakeholders.
📌 Required Qualifications
5+ years in Machine Learning Engineering, with specific experience in leading technical projects or mentoring engineers.
Exceptional ability to distill complex technical topics (e.g., compute complexity, infrastructure costs) into clear, business-relevant insights for decision-makers.
Proven experience in optimizing ML pipelines and infrastructure. Familiarity with tools like MLflow, Kubeflow, Airflow, and Data Versioning tools (DVC, etc.).
Proficient in Python, Docker, and Kubernetes. You treat ML models as software artifacts that need testing and version control.
Experience with Transformer architectures, Embeddings, and LLM fine-tuning. Familiarity with frameworks like PyTorch, Hugging Face, and vLLM.
Experience processing or fine-tuning models for multi-lingual environments.
⭐ Desirable Experience
Experience working with Multimodal models (Image-to-Text, Text-to-Image, VLMs like CLIP, LLaVA).
Understanding of GenAI security threats (e.g., Prompt Injection).
Experience optimizing inference speed (quantization, distillation, vLLM) for real-time applications.
Experience with Vector DBs for RAG applications.
🎁 Benefits
Innovative Environment: Be part of a company at the forefront of technology to provide security in GenAI, with opportunities to work on groundbreaking projects.
Growth Opportunities: Take your career to new heights with our career development programs and growth-focused culture.
Dynamic Team: Join a multi-cultural and dynamic team of dedicated professionals who inspire and support each other.
Compensation: Competitive salary and benefits package, commensurate with experience and performance.
