Flex

Senior Staff AI/Machine Learning Engineer

Flex

full-time

Posted on:

Origin:  • 🇺🇸 United States

Visit company website
AI Apply
Manual Apply

Salary

💰 $221,000 - $237,000 per year

Job Level

Senior

Tech Stack

AirflowAWSAzureCloudGoogle Cloud PlatformKubernetesPandasPythonPyTorchScikit-LearnSparkTensorflow

About the role

  • Lead the design, development, and deployment of scalable machine learning models for high-impact fintech applications
  • Drive 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 business challenges into ML solutions
  • Build, maintain, and optimize ML systems meeting performance, scalability, and reliability standards in production
  • Continuously monitor and enhance model performance through experimentation and tuning
  • 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 and frameworks
  • Champion responsible AI practices ensuring explainability, fairness, bias mitigation, and auditability

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
  • Excellent analytical and problem-solving skills and ability to collaborate across cross-functional teams