Staff AI Engineer
Vasion
AirflowAWSAzureCloudDockerGoogle Cloud PlatformGraphQLGRPCKubernetesPyTorchTensorflow
Senior Staff AI/Machine Learning Engineer
Flex
full-time
Posted on:
Location: 🇺🇸 United States
Visit company websiteSalary
💰 $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
