Apply on Employer Site

Flex · 5 hours ago

Sr. Staff AI/Machine Learning Engineer

United States

Full-time

Remote

Senior Level, Lead/Staff

$209K/yr - $225K/yr

6+ years exp

Flex is a growth-stage FinTech company headquartered in NYC, dedicated to transforming the rent payment experience for users. They are seeking a Senior Staff AI/Machine Learning Engineer to lead the development of advanced AI and machine learning systems that enhance business growth and product offerings.

AppsBillingFinancial ServicesFinTechSoftware

H1B Sponsor Likely

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

Qualification

Machine LearningPythonCloud PlatformsML FrameworksDistributed ComputingData Workflow ToolsPerformance OptimizationAnalytical SkillsProblem-SolvingCollaboration

Required

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

Benefits

Competitive medical, dental, and vision available from Day 1

Company equity

401(k) plan with company match (our company match kicks off at the beginning of 2026)

Unlimited paid time off + 13 company paid holidays

Parental leave

Flex Cares Program

Free Flex subscription

Competitive compensation + company equity

Unlimited PTO

Company

Flex

Glassdoor3.9

Flex enables people to pay rent bills on a schedule that fits their personal finances.

Founded in 2019

New York, New York, USA

201-500 employees

https://getflex.com

H1B Sponsorship

Flex has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (1)

2024 (1)