A

AIFT

Cybersecurity and insurance for AI ecosystems

Machine Learning Engineer Lead

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeNo salary listedIn-person

About the Job

Requirements

  • At least 5 years of experience in Machine Learning Engineering, including experience leading technical projects or mentoring engineers.
  • Ability to distill complex technical topics, such as compute complexity and infrastructure costs, into clear, business-relevant insights for decision-makers.
  • Proven experience optimizing machine learning pipelines and infrastructure, with familiarity with MLflow, Kubeflow, Airflow, and data-versioning tools such as DVC.
  • Proficiency in Python, Docker, and Kubernetes, and an approach that treats machine learning models as software artifacts requiring testing and version control.
  • Experience with Transformer architectures, embeddings, and large language model fine-tuning; familiarity with PyTorch, Hugging Face, and vLLM.
  • Experience processing or fine-tuning models for multilingual environments.

Responsibilities

  • Collaborate with the Security Research Team to operationalize new threat-detection techniques, determining model architecture and training strategy.
  • Lead fine-tuning of language models, including with LoRA/PEFT, to optimize for supported multilingual languages and security intents.
  • Prepare for multimodal text, image, and audio capabilities by evaluating and implementing models to detect visual prompt injections and non-textual threats as the product evolves.
  • Take ownership of existing machine-learning pipelines and optimize and scale CI/CD/CT workflows to improve training efficiency and deployment velocity.
  • Implement and enforce data-versioning strategies, such as DVC, to ensure reproducibility of model artifacts and datasets.
  • Monitor model drift and performance to maintain reliability in a production security environment.
  • Work with the Platform Engineering Team to integrate machine-learning models into the product architecture and ensure interaction between model-inference services and platform logic.
  • Lead and mentor Machine Learning Engineers, promoting engineering rigor, code quality, and operational excellence.
  • Manage GPU resources and compute budgets for training and inference workloads.
  • Explain machine-learning concepts, including FLOPS, quantization trade-offs, and model latency versus accuracy, to executive leadership and clients.
  • Justify compute-resource investments and explain trade-offs between infrastructure costs, including GPU hours, and performance gains to non-technical stakeholders.

Desired Qualifications

  • Experience working with multimodal models, including image-to-text, text-to-image, and vision-language models such as CLIP and LLaVA.
  • Understanding of generative-AI security threats, such as prompt injection.
  • Experience optimizing inference speed through quantization, distillation, or vLLM for real-time applications.
  • Experience with vector databases for retrieval-augmented generation applications.

About the Company

AIFT

Cybersecurity and insurance for AI ecosystems

AIFT is a Cayman Islands–based holding group that provides cybersecurity and insurance protection tailored to AI, Web3, and other digital mega trends. Its offerings combine tech-enabled security services with insurance designed for digital-era risks, using AI and other digital trends to protect clients. It differentiates itself by targeting underserved market segments, moving quickly to establish a presence, and uniting multiple risk-management offerings under one umbrella for faster growth. Its goal is to become the global security layer for AI, Web3, and next-generation digital lifestyles, pursuing scalable growth and profitability, including achieving its first operating profit in Q3 2024.

Company size
51-200
Company stage
Not listed
Total funding
Not listed
Headquarters
Taiwan
Founded
2016

Recently Posted Jobs

Explore more opportunities at AIFT

See more jobs