Full-Time

Sr. Staff AI/Machine Learning Engineer

Posted on 9/5/2025

Flex

Flex

10,001+ employees

Global contract manufacturer offering design engineering

Compensation Overview

$221k - $237k/yr

+ Base salary for NY: 221000-237000 + Base salary for all other states: 209000-225000

New York, NY, USA

Hybrid

Category
AI & Machine Learning (3)
, ,
Required Skills
Scikit-learn
Kubernetes
Microsoft Azure
Python
Airflow
Tensorflow
Keras
Pytorch
Apache Spark
AWS
Pandas
Google Cloud Platform
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.
Desired Qualifications
  • Experience with FinTech systems and regulatory considerations is a plus.
  • Experience with large-scale data processing and real-time inference.
  • Familiarity with database systems and data warehousing concepts.
  • Strong communication and leadership skills, mentoring junior engineers and collaborating with cross-functional teams.

Flex is a global manufacturing partner that designs, engineers, manufactures, and manages the supply chain for products across diverse industries. Its offering starts with product design and engineering, followed by manufacturing and testing, and extends to sourcing, logistics, and end-to-end delivery of finished goods. Unlike many players that focus on a single niche, Flex provides end-to-end, globally integrated services across multiple industries, leveraging a large worldwide network and scalable operations. The company’s goal is to help customers bring products to market quickly and efficiently by handling development, manufacturing, and supply chain management at scale.

Company Size

10,001+

Company Stage

IPO

Headquarters

Singapore, Singapore

Founded

1969

Simplify Jobs

Simplify's Take

What believers are saying

  • Cloud and Power Infrastructure segment surges 31% in fiscal 2026, targeting 65-75% FY2027 growth.
  • Q1 2026 revenue hits $7.48 billion, beating estimates by 7.5%, with FY2027 guidance at $32.3-33.8 billion.
  • €90 million Hungary facility supplies EV components to BMW, Ford, Volkswagen, creating 210 jobs.

What critics are saying

  • Q1 2027 spinoff removes Cloud and Power Infrastructure, cutting revenue growth to single digits.
  • Analyst targets average $76.60 versus $132 stock price, causing 40-50% drop in 6-12 months.
  • Hyperscalers like NVIDIA in-house 800 VDC racks, eliminating 30% of FY2027 revenue in 12-18 months.

What makes Flex unique

  • Flex delivers NVIDIA Omniverse DSX Blueprint with 800 VDC power racks for AI data centers.
  • Flex manufactures AMD Instinct AI platforms in Austin, Texas, supporting U.S. localization.
  • Flex evolved from 1969 Silicon Valley PCB soldering to end-to-end AI infrastructure solutions.

Help us improve and share your feedback! Did you find this helpful?

Benefits

Health Insurance

Dental Insurance

Vision Insurance

401(k) Company Match

Company Equity

Unlimited Paid Time Off

Paid Holidays

Parental Leave

Profit Sharing

Company News

CityBiz
Mar 30th, 2026
Flex To Acquire Electrical Power Products (EP²)

Flex (NASDAQ: FLEX) announced today that it has entered into a definitive agreement to acquire Electrical Power Products, Inc., a leading... Read More

Yahoo Finance
Mar 25th, 2026
US tech stocks watch: Palantir, Reddit among top growth picks as earnings forecast to rise 16%

Flex, a technology solutions provider, is expanding its AI infrastructure capabilities through a collaboration with NVIDIA to develop the 800 VDC Power Rack, which promises up to 30% faster deployment through modular designs. The company is also partnering with AMD to manufacture the AMD Instinct platform in Texas, aligning with domestic production trends. With a market capitalisation of approximately $23.91 billion, Flex generates revenue through its Flex Agility Solutions and Flex Reliability Solutions segments, contributing $14.78 billion and $12.06 billion respectively. The company's annual earnings are forecast to grow by 21.7%, whilst revenue growth is projected at 6.2% per year. These strategic moves position Flex to capitalise on growing demand for AI infrastructure and localised manufacturing.

Evertiq
Mar 24th, 2026
Flex breaks ground on €90 million facility in Hungary.

Flex breaks ground on €90 million facility in Hungary. Flex has laid the foundation stone for a new warehouse in zalaegerszeg, Hungary, marking a further milestone in its EUR 90 million investment project in the country. The development includes both a manufacturing plant and a warehouse, and will focus on the production of components for electric and hybrid vehicles. The project is expected to create around 210 new jobs, including approximately 20 positions in research and development. The new facility will supply components to OEMs such as BMW, Ford and Volkswagen, as well as Tier 1 suppliers. Flex operates in 30 countries with more than 100 sites, providing design and manufacturing services across multiple industries, including automotive, industrial, and consumer sectors. According to the Hungarian Investment Promotion Agency, Flex primarily cooperates with domestic suppliers in Hungary, contributing to the competitiveness of the local supply chain and the broader regional economy. The company also maintains long-term partnerships with academic institutions, including the University of Pannonia and Budapest University of Technology and Economics, particularly through dual education programs. Explore the evolving landscape of the electronics industry and plan your 2026 season with Evertiq Expo. Evertiq New Media AB will begin the year on March 26 in Tampere, Finland, and on April 23 inaugurate its first-ever Evertiq Expo in Zurich. Each event brings together key industry players, researchers and innovators shaping the future of electronics. To stay up to date with the latest developments, sign up for the Newsletter.

Yahoo Finance
Mar 17th, 2026
Flex unveils NVIDIA-aligned AI factory designs with 800 VDC power and liquid cooling for hyperscale data centers

Flex announced new NVIDIA Omniverse DSX Blueprint reference designs on 16 March 2026, offering prefabricated, modular power and liquid-cooling systems for AI data centres. The systems feature an 800 VDC Power Rack and support phased migration from traditional AC setups, positioning Flex as a systems-level partner for hyperscale AI infrastructure rather than a traditional contract manufacturer. The announcement follows Flex's expanded collaboration with AMD to build Instinct AI platforms in Austin. Together, these moves underscore Flex's strategy to become a full-stack AI hardware and infrastructure partner, though customer concentration and potential in-housing of power and cooling capabilities remain key risks. Optimistic analysts project Flex reaching $29.7 billion revenue and $1.4 billion earnings by 2028. The company's shares rose 5.1% following the announcement.

Plant Services
Mar 17th, 2026
It's just business: manufacturing moves from HP, Flex, AIM MRO and more.

It's just business: manufacturing moves from HP, Flex, AIM MRO and more. A recent look at recent manufacturing business deals, including plans to make the world's largest tofu factory. March 17, 2026 The business of manufacturing never stops. Industrial companies merge, invest and strategically expand to keep competitive, even under adverse business conditions. Read on for more details on five such expansions and acquisitions made by industrial and manufacturing companies. Partnership: HP and Volkmann announced March 3 that the companies would collaborate on a contained metal powder handling solution for metal jet printers. Under the terms of the deal, HP's Metal Jet S100 3D printers will be compatible with specially-designed proprietary Volkmann contained metal powder management systems. In a statement, Alexandre Tartas, a global head of sales at HP Additive Manufacturing Solutions, said the containment solution would improve device efficiency and safety. Expansion: Nasoya announced March 11 it would spend $55 million to expand an existing tofu manufacturing plant in Ayer, Massachusetts. The expansion will add 65,000 square feet of new space and increasing the plant workforce by 20%, for a total size of 200,000 square feet and daily capacity of 400,000 pounds of tofu. The company says that, after the expansion, the tofu factory will be the largest in the world. Partnership: Flex announced March 2 it would expand its manufacturing collaboration with AMD to produce the semiconductor manufacturer's AMD Instinct graphics processing unit platform, which are experiencing a surge in demand alongside the use of GPUs in data centers for AI services. According to a release from Flex, the contract manufacturer has already started manufacturing the AMD Instinct in its Austin, Texas headquarters with plans to increase production. Expansion: Yazoo Mills announced March 12 that it had finished construction of its third paper tubes and cores factory in Hanover, Pennsylvania. According to a company statement, the $14 million plant construction included relocating eight winding lines and 14 recutting machines to the new site. The plant is Yazoo Mills' third manufacturing location.

INACTIVE