Sr. Software Engineer, Machine Learning
- San Francisco, CA
- Position open to remote: Yes
We are a small team who is part of Square’s Platform Infrastructure Engineering organization. As an engineer on our team, you’ll work on the machine learning tools and frameworks used by Square’s data scientists and Machine Learning Engineers (MLEs). We partner closely with data scientists and MLEs across all lines of businesses to understand their needs and develop solutions that help them move faster.
We support multiple frameworks that make data extraction, preparation, and model training easier. One such framework is called Bionic which we designed to help MLEs iterate more rapidly on their research and help them share their code effortlessly. We already open sourced this framework and plan to continue contributing to the open source community with our future projects.
We also plan to help improve engineering productivity by making it easier to set up and maintain development environments for data scientists and MLEs. We want Python, which is the language of choice for building and deploying machine learning models and is used heavily within our machine learning ecosystem, to gain parity with other core languages. You will lead the promotion of Python as a fully-supported programming language at Square.
Strong software engineering background with emphasis in data science and machine learning. Relevant graduate degree or equivalent industry experience (5+ years preferred).
Enough (and recent) proficiency with Python, R, Java and other relevant tools that will allow you to support and mentor the team.
Experience with machine learning libraries/frameworks like TensorFlow, PyTorch, Keras, Pandas, NumPy, Scikit-learn or XGBoost.
Experience leading projects and mentoring others to solve problems using machine learning.
Experience with cloud computing platforms is preferred (i.e. AWS or Google Cloud Platform)
Technologies we use
Python, Java, Google Cloud Platform, AWS, Snowflake
Python machine learning stack (pandas, scikit-learn, Jupyter, etc.)
MySQL, Redis, Hibernate, jOOQ, Bigtable
At Square, we want you to be well and thrive. Our global benefits package includes:
- Healthcare coverage
- Retirement Plans
- Employee Stock Purchase Program
- Wellness perks
- Paid parental leave
- Paid time off
- Learning and Development resources