Machine Learning Engineer, Knowledge Platform

  • New York, NY
  • Full-time

Company Description

We believe the economy is better when everyone has access. When everyone has room to grow. No one should be left out because the cost is too great or the technology too complex. We started with a little white credit card reader but haven’t stopped there. We’re empowering the independent electrician to send invoices, setting up the favorite food truck with a delivery option, helping the ice cream shop pay its employees, and giving the burgeoning coffee chain capital for a second, third, and fourth location. We’re here to help sellers of all sizes start, run, and grow their business—and helping them grow their business is good business for everyone.

Job Description

The knowledge platform team is seeking senior machine learning engineers to join our engineering team.  Square is seeking senior machine learning engineers to build a company-wide state-of-the-art knowledge platform. Our goal is to provide tooling, infrastructure, and guidance to unify and level up the 50-100 engineers and data scientists working on ML at Square and work on advanced ML solutions that are applicable across the company.  The ideal candidate will have industry experience in solving and optimizing large-scale machine learning problems.  We are looking for passionate and self-driven innovators to help us build this V1 platform from the ground up and be part of a fast-paced team.  You will be expected to contribute in building ML systems/tooling and building advanced ML models that scale. You will have a wide impact across the company with opportunity to publish papers, contribute to open-source, influence and collaborate with the data science community across Square.


You have:

  • Experience in many of the following areas is highly desired - AutoML, Knowledge Graphs, recommendation systems, NLP and AI Agents.
  • Experience with cloud computing platforms, such as AWS, Google Cloud or Azure.
  • Experience in designing and productionizing large-scale distributed systems built around machine learned models and big data.
  • Experience or familiarity with interpretable machine learning is a plus.
  • Ability to produce scalable and robust production-quality code incorporating testing, evaluation, and monitoring.
  • An advanced degree (PhD or MS) in Computer Science

Technologies we use:

  • Java, Python, Google Cloud Platform, AWS, Snowflake
  • Python ML stack (pandas, scikit-learn, etc.)

Additional Information

At Square, our purpose is to empower – within and outside of our walls. In order to build the best tools for the businesses and customers we support all over the world, we have to start at home with a workforce as diverse and empowered as our sellers. To this end, we take great care to evaluate all employees and job applicants equally, based on merit, competence, and qualifications. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, disability status, or any other characteristic protected by law. We encourage candidates from all backgrounds to apply and always consider qualified applicants with arrest and conviction records, in accordance with the San Francisco Fair Chance Ordinance. Applicants in need of special assistance or accommodation during the interview process or in accessing our website may contact us by sending an email to assistance(at) We will treat your request as confidentially as possible. In your email, please include your name and preferred method of contact, and we will respond as soon as possible. Perks At Square, we want you to be well and thrive. Our global benefits package includes: Healthcare coverage, Retirement Plans, Employee Stock Purchase Program, Meal reimbursements, Wellness perks, Paid parental leave, Flexible time off, Learning and Development resources