Staff Artificial Intelligence Engineer, Conversations

Square Messages is a messaging platform for merchants to manage conversations with their customers across multiple channels (SMS, email, etc.). Situated within the Square ecosystem, Square Messages has integrations with various Square products that enrich messages with appointment bookings, receipts, invoices, coupons, marketing, and more.

Messages is also home to Square Assistant – an AI-powered virtual assistant to help merchants serve their customers 24/7. Square Assistant responds to customers asking for help with their appointments, including cancellations, confirmations, and conversational rescheduling of appointments. For more nuanced customer requests, Square Assistant saves merchants time by utilizing an in-house large language model to suggest responses, as well as actions that integrate with the rest of the Square Ecosystem. In the year ahead, we plan to expand Square Assistant’s capabilities in many ways, like personalizing responses for each merchant or deploying larger language models in production.

We are looking for Staff AI Engineers with interests in language modeling, conversational AI, and production machine learning at scale.

You will...

• Play a key role in designing and developing novel conversational AI systems within Square.
• Partner with product managers, designers, and the other engineering teams to design, iterate, and ship new product experiences.
• Influence and lead the team’s future roadmap.
• Ship cutting-edge AI to businesses that traditionally haven’t had access to it.
• Train and deploy state of the art language models.
• Build and maintain AI infrastructure that handles millions of conversations.

Qualifications

You have…

• 10+ years of experience with Machine Learning.
• A track record of leading the design and execution of production-level machine learning systems.
• Experience influencing and shaping team’s technical roadmap through close collaboration with stakeholders and driving execution.
• Experience with Large Language Models.
• Experience programming with PyTorch in a production environment.
• Experience with Python. Openness to programming in Java.
• A commitment to writing clean and maintainable code.
• Eagerness to share your own ideas, and openness to those of others.

Full-time employee benefits include the following:

• Healthcare coverage (Medical, Vision and Dental insurance)
• Health Savings Account and Flexible Spending Account
• Retirement Plans including company match
• Employee Stock Purchase Program
• Wellness programs, including access to mental health, 1:1 financial planners, and a monthly wellness allowance
• Paid parental and caregiving leave
• Paid time off (including 12 paid holidays)
• Paid sick leave (1 hour per 26 hours worked (max 80 hours per calendar year to the extent legally permissible) for non-exempt employees and covered by our Flexible Time Off policy for exempt employees)
• Learning and Development resources
• Paid Life insurance, AD&D, and disability benefits

These benefits are further detailed in Block's policies. This role is also eligible to participate in Block's equity plan subject to the terms of the applicable plans and policies, and may be eligible for a sign-on bonus. Sales roles may be eligible to participate in a commission plan subject to the terms of the applicable plans and policies. Pay and benefits are subject to change at any time, consistent with the terms of any applicable compensation or benefit plans.