[Remote] Sr. Staff AI/Machine Learning Engineer at Flex

United States

Flex Logo
Not SpecifiedCompensation
Senior (5 to 8 years), Expert & Leadership (9+ years)Experience Level
Full TimeJob Type
UnknownVisa
FinTechIndustries

Requirements

  • Master’s or Ph.D. in Computer Science, Engineering, Statistics, or a related technical field
  • 6+ years of experience in machine learning, including 3+ years in senior or staff-level roles
  • Expert proficiency in Python (or similar languages) and experience with key ML libraries and frameworks such as TensorFlow, PyTorch, and scikit-learn
  • Extensive experience with cloud platforms (AWS, GCP, Azure), ML infrastructure tools (e.g., SageMaker, Vertex AI, MLflow), distributed computing (e.g., Spark, Kubernetes), and data workflow tools (e.g., Pandas, Airflow)
  • Proven track record of designing, building, and deploying end-to-end ML systems at scale in a production environment
  • Strong background in performance optimization, version control, CI/CD, and managing the ML model lifecycle in real-world applications
  • Excellent analytical and problem-solving skills, with a proven ability to collaborate across cross-functional teams to deliver production-ready solutions that meet business goals

Responsibilities

  • Lead the design, development, and deployment of scalable machine learning models for high-impact fintech applications, driving the full ML lifecycle—from research and data exploration to training, evaluation, deployment, and monitoring
  • Collaborate with data scientists, engineers, and product teams to translate complex business challenges into machine learning solutions that directly support product and business objectives
  • Build, maintain, and optimize robust ML systems that meet stringent performance, scalability, and reliability standards in a production environment
  • Continuously monitor and enhance model performance through experimentation, hyperparameter tuning, and iteration to ensure alignment with evolving business needs
  • Utilize distributed computing frameworks and cloud-native platforms to develop and maintain data pipelines, feature stores, model serving infrastructure, and MLOps workflows
  • Evaluate, adopt, and implement state-of-the-art AI/ML tools, frameworks, and methodologies to ensure technical excellence and a competitive advantage
  • Champion responsible AI practices by ensuring models comply with regulatory and ethical standards, with a focus on explainability, fairness, bias mitigation, and auditability
  • Stay current with emerging trends and innovations in AI, machine learning, software engineering, and scalable deployment to keep our systems on the cutting-edge

Skills

Key technologies and capabilities for this role

Machine LearningAIML ModelsModel DeploymentScalable ML SystemsHyperparameter TuningModel MonitoringData ExplorationModel TrainingModel EvaluationFintech ApplicationsProduction ML

Questions & Answers

Common questions about this position

What education is required for this Senior Staff AI/ML Engineer role?

A Master’s or Ph.D. in Computer Science, Engineering, Statistics, or a related technical field is required.

What experience level is needed for this position?

Candidates need 6+ years of experience in machine learning, including 3+ years in senior or staff-level roles.

What programming skills are required?

Expert proficiency in Python (or similar languages) is required.

Is this role remote or based in a specific location?

This information is not specified in the job description.

What is the salary or compensation for this position?

This information is not specified in the job description.

Flex

Comprehensive contract manufacturer for electronics

About Flex

Flex provides comprehensive electronics manufacturing services, originally starting with circuit board production for tech companies in Silicon Valley. The company automates its manufacturing processes to ensure reliability and cost-effectiveness, which has made it a trusted partner for various industries, including automotive, healthcare, and consumer electronics. Flex's services encompass design, engineering, manufacturing, and supply chain management, allowing businesses to outsource their production needs and focus on their core activities. A key differentiator for Flex is its subsidiary, Anord Mardix, which specializes in critical power solutions and custom products for clients with complex requirements. Additionally, Flex is dedicated to sustainability, aiming for a significant portion of its customers to adopt science-based emissions targets by 2025, reflecting its commitment to environmental responsibility.

Singapore, SingaporeHeadquarters
1969Year Founded
$200MTotal Funding
IPOCompany Stage
Industrial & Manufacturing, EnergyIndustries
10,001+Employees

Benefits

Health Insurance
Dental Insurance
Vision Insurance
401(k) Company Match
Company Equity
Unlimited Paid Time Off
Paid Holidays
Parental Leave
Profit Sharing

Risks

Integrating JetCool Technologies may lead to operational and cultural alignment challenges.
Partnership with Musashi Energy exposes Flex to risks of emerging technology failures.
Jamdeo joint venture faces competitive pressures in the smart TV and media streaming market.

Differentiation

Flex offers end-to-end manufacturing solutions, from design to supply chain services.
Anord Mardix, a Flex subsidiary, specializes in critical power solutions and custom offerings.
Flex's commitment to sustainability aligns with global efforts to combat climate change.

Upsides

Flex's acquisition of JetCool enhances its data center cooling solutions for AI servers.
Partnership with Musashi Energy boosts Flex's AI data center energy storage capabilities.
Flex's Zhuhai facility's AME Excellence Award highlights its world-class manufacturing standards.

Land your dream remote job 3x faster with AI