Machine Learning Engineer

Job Summary

To enhance the capabilities of the AI Guard team, the full-time remote Staff Machine Learning Engineer will design, build, and deploy end-to-end machine learning pipelines while integrating advanced AI functionalities into production-ready SaaS offerings.

Key responsibilities

• Build and deploy end-to-end ML pipelines including data curation, training, evaluation, and serving
• Develop deep learning models for security use cases, focusing on high-volume traffic data
• Optimize models for production using techniques such as quantization and batching across cloud platforms

Required qualifications

• Bachelor's or advanced degree in Computer Science, Machine Learning, Mathematics, Physics, Statistics, Engineering, or a related field, with 2+ years of applied ML experience
• Demonstrated experience with modern AI/ML frameworks and foundational model workflows
• Strong Python programming skills and hands-on experience with deep learning frameworks like PyTorch, TensorFlow, or JAX
• Proven experience training and deploying large language models and deep learning models in production environments
• Solid understanding of machine learning fundamentals, including optimization and evaluation metrics