Senior AI/ML Engineer
Senior AI/ML Engineer building transformer and deep learning models on financial and behavioral data to power personalized growth and marketing experiences at Chime. Requires strong production ML experience with PyTorch, AWS, and large-scale data infrastructure.
About the job
Responsibilities
- Develop and deploy sequential deep learning models and traditional machine learning systems to power growth and marketing initiatives
- Build predictive models using large-scale financial, transactional, and behavioral datasets to improve personalization and member engagement
- Partner cross-functionally with Growth & Marketing, Product, and Engineering teams to drive strategic AI/ML initiatives
- Design and improve infrastructure for training, serving, and monitoring large-scale ML and deep learning systems in both batch and real-time environments
- Generate insights and recommendations that improve growth effectiveness and the overall member experience
- Contribute to experimentation frameworks, optimization strategies, and scalable ML platform capabilities
- Help identify technology gaps and opportunities where AI/ML solutions can create measurable business impact
Requirements
- Deep expertise in building sequential and deep learning models, especially outside of traditional NLP applications and within financial or behavioral data domains
- Strong experience across the end-to-end ML lifecycle, including training, experimentation, optimization, deployment, and monitoring
- Experience working with large-scale transactional, financial, or behavioral datasets to develop predictive models
- Hands-on experience with AWS and modern ML infrastructure tools such as SageMaker, Kafka, Airflow, Redis, Snowflake, and Spark
- Strong proficiency in Python, SQL, and distributed computing and model training frameworks such as PyTorch and PySpark for scalable ML development
- A strong MLOps mindset with experience deploying and maintaining production-grade ML systems
- The ability to operate independently, collaborate cross-functionally, and move quickly in ambiguous environments
Compensation & Benefits
- Base salary: $172,000 - $238,000
- Eligible for bonus, competitive equity package, and benefits
- 401k match plus medical, dental, vision, life, and disability benefits
- Generous vacation policy and company-wide paid days off
- Up to 24 weeks of paid parental leave for birthing parents and 12 weeks for non-birthing parents
- Annual wellness stipend
- 1% of time off to support local community organizations
Skills
Python, SQL, PyTorch, Pyspark, AWS, SageMaker, Kafka, Airflow, Redis, Snowflake, Spark, Deep Learning, Machine Learning, MLOps
Similar jobs
ML Engineering jobs
Instacart
United StatesBuild and deploy machine learning systems that apply economic theory, econometrics, and causal inference to marketplace problems. The role requires advanced training in economics, strong Python and data skills, and production ML experience for senior-level hires.
Everlaw
Oakland, CA
Builds and operates AI platform capabilities including RAG pipelines, semantic retrieval, agentic orchestration, and LLM integrations to power legal tech products. Requires 4+ years in distributed cloud systems, AI/ML experience, and proficiency in modern programming languages.
Beacon Biosignals
United States
The Senior Algorithm Engineer leads development and production deployment of machine and deep learning algorithms for biosignal and medical-device applications. The role requires 5+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated health or similar domains.
Ai2
Seattle, WA
Senior Research Engineer tailoring and deploying machine learning models for partner applications across geospatial and environmental domains. The role requires PyTorch expertise, end-to-end ML deployment experience, geospatial tools knowledge, and strong independent execution.
Swayable
New York, NYSenior engineer developing and productizing AI, machine learning, scientific computing, and data-analysis capabilities for a high-performance analytics engine. Requires 5+ years building quantitative data-intensive software and expertise in Python, machine learning, scalable architecture, and distributed computing.