Staff Artificial Intelligence Engineer, Conversations

Key Skills

Aartificial intelligence, Deep Learning, Java Programming, Large Language Models - LLMs, Machine learning techniques, Python Programming, PyTorch

Job Description

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 senior 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.
• Train and deploy state of the art language models.
• Build and maintain AI infrastructure that handles millions of conversations.
• Collaborate with product managers, designers, and the other engineering teams to design, iterate, and ship new product experiences.
• Work cross-functionally with other Square teams to enhance the Messages platform integrations.
• Ship cutting-edge AI to businesses that traditionally haven’t had access to it.

Qualifications

You have…

• 6+ years of experience with Machine Learning.
• Experience with Large Language Models OR comparable Deep Learning experience.
• Experience with PyTorch is strongly recommended.
• Experience with Python. Openness to programming in Java.
• A track record of designing and maintaining production-level machine learning systems.
• A commitment to writing clean and maintainable code.
• Interest in helping to shape the mission and roadmap of the team, and rallying others to join you.
• Eagerness to share your own ideas, and openness to those of others.

Additional Information

Block takes a market-based approach to pay, and pay may vary depending on your location. U.S. locations are categorized into one of four zones based on a cost of labor index for that geographic area. The successful candidate’s starting pay will be determined based on job-related skills, experience, qualifications, work location, and market conditions. These ranges may be modified in the future.

Zone A: USD $202,500 - USD $247,500

Zone B: USD $192,400 - USD $235,200

Zone C: USD $182,300 - USD $222,800

Zone D: USD $172,200 - USD $210,400

To find a location’s zone designation, please refer to this resource. If a location of interest is not listed, please speak with a recruiter for additional information.

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.

We’re working to build a more inclusive economy where our customers have equal access to opportunity, and we strive to live by these same values in building our workplace. Block is a proud equal opportunity employer. We work hard to evaluate all employees and job applicants consistently, without regard to race, color, religion, gender, national origin, age, disability, veteran status, pregnancy, gender expression or identity, sexual orientation, citizenship, or any other legally protected class.