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

United States

Flex Logo
Not SpecifiedCompensation
N/AExperience Level
N/AJob Type
Not SpecifiedVisa
N/AIndustries

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
  • Collaborate with data scientists, engineers, and product teams to translate complex business challenges into machine learning solutions
  • 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
  • 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
  • Champion responsible AI practices by ensuring models comply with regulatory and ethical standards
  • Stay current with emerging trends and innovations in AI, machine learning, software engineering, and scalable deployment

Skills

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