Summary

We are looking for a Senior Machine Learning Engineer to join the Machine Learning Features team within Block's Machine Learning Platform. This team is responsible for developing and maintaining the systems that compute and serve millions of ML features every day, powering Machine Learning models across Block.

As a senior engineer, you will help scale our high-demand ML feature pipelines and services to meet the needs of teams across Cash App, Square, Afterpay, and more. You will also contribute to the design and development of a groundbreaking new ML feature system, aimed at redefining how Machine Learning features are computed and delivered at scale.

At Block, you'll be part of a collaborative, forward-thinking team that is shaping the future of Machine Learning infrastructure. The systems you build will directly empower ML teams across multiple business units, impacting millions of users worldwide. If you're excited about working at the intersection of large-scale distributed systems and Machine Learning, we'd love to hear from you.

You Will

1. Own and maintain mission-critical ML feature computation and serving systems that support millions of daily feature requests
2. Design and build the next-generation ML feature platform to enable faster iteration and higher-quality ML models across Block
3. Partner closely with Data Scientists, ML Modelers and Software Engineers to deliver reliable, performant, and scalable systems
4. Drive architectural decisions, system reliability improvements, and infrastructure automation
5. Mentor and guide other engineers on best practices in large-scale distributed systems, feature engineering, and ML infrastructure
6. Champion engineering excellence through code reviews, technical documentation, and continuous improvement

You Have

1. 5+ years of experience in software engineering, with at least 3+ in large-scale data or ML infrastructure
2. Strong proficiency in programming languages such as Java, Python, Kotlin or Go
3. Experience building and operating distributed data systems at scale (e.g., Spark, Flink, Kafka, Databricks, Snowflake)
4. Proven ability to design for high performance, scalability, and reliability
5. Deep knowledge of cloud infrastructure (AWS, GCP) and containerized systems (Kubernetes, Docker)
6. Strong collaboration skills and ability to work cross-functionally with ML practitioners and infrastructure engineers
7. Experience mentoring junior engineers and driving technical direction on complex projects

Nice to have

1. Familiarity with ML workflows, platforms, and systems
2. Experience building or maintaining ML batch or real-time feature systems
3. Background in MLOps, experimentation platforms, or ML observability

Skills

• AWS
• Communications Skills
• Development
• Java
• Machine Learning
• Python
• Software Engineering
• Team Collaboration