Staff Machine Learning Engineer

Summary

Join Flex, a growing FinTech company in NYC, as a Staff Machine Learning Engineer. You will play a key role in developing and deploying cutting-edge machine learning systems. This position requires expertise in building and deploying ML models in production environments, using various tools and technologies. You will collaborate with data scientists, product teams, and engineers. The role offers a competitive salary and a comprehensive benefits package. Flex is committed to building an inclusive culture and offers a remote-friendly work environment.

Requirements

• Master’s or Ph.D. in Computer Science, Engineering, or a related field
• 6+ years of experience as a Machine Learning Engineer, with expertise in building and deploying machine learning models in production environments
• Strong proficiency in Python, or similar programming languages, and experience with ML libraries like TensorFlow, PyTorch, and scikit-learn
• Extensive experience with cloud platforms (e.g., AWS, GCP, Azure) and distributed computing frameworks (e.g., Spark, Kubernetes)
• Proven track record of implementing end-to-end machine learning pipelines, from data preprocessing to production deployment and monitoring
• Strong background in model optimization, version control, and CI/CD practices for machine learning
• Excellent problem-solving abilities and the capacity to collaborate with cross-functional teams to deliver high-quality, production-ready systems

Responsibilities

• Own the end-to-end lifecycle of machine learning projects, from data collection and preprocessing to model deployment, monitoring, and maintenance in a production environment
• Build, maintain, and optimize robust data pipelines that support model development, training, and deployment at scale
• Implement machine learning algorithms and models that meet performance, scalability, and reliability requirements in a production setting
• Collaborate with data scientists, engineers, and product teams to design and deploy machine learning systems that address business and product needs
• Continuously monitor and improve model performance, conducting experiments, tuning hyperparameters, and ensuring models meet business objectives
• Leverage distributed computing frameworks and cloud-based platforms to process large-scale datasets efficiently
• Stay up-to-date with the latest advancements in machine learning, software engineering practices, and deployment strategies to keep our systems cutting-edge

Benefits

• Competitive pay
• 100% company-paid medical, dental, and vision
• 401(k) + company equity
• Unlimited paid time off with a PTO minimum + 13 company paid holidays
• Parental leave
• Flex Cares Program: Non-profit company match + pet adoption coverage
• Free Flex subscription
• Competitive Pay
• Company Equity
• Unlimited PTO