Software Architect - AI/ML

Key Skills

Aartificial intelligence, AWS, GPUs, JAX framework, Large Language Models - LLMs, Leadership Skill, Machine learning techniques, Optimization, PaaS, PyTorch, Scrum Agile Methodology

Job Description

This position require top-notch technical skills and outstanding analytical and influencing skills paired with extraordinary business insight. It is a multi-functional role that requires alignment building and communication within a large engineering organization. Software architects at Heroku write code.

Your Impact:

• Coordinate and drive the end-to-end architecture of AI, ML, and LLM capabilities across the Heroku platform
• Craft overall architectural principles, frameworks, and standards
• Create solutions that are enterprise scalable and maintainable
• Drive strategic technology initiatives, working with product managers, other architects, engineers, subject-matter experts, and the broader Salesforce organization
• Work in a consultative fashion to improve communication, teamwork and alignment among teams inside and outside of the organization
• Analyze and provide feedback on product strategy
• Provide input into long-range product requirements, standard methodologies, and operational guidelines, with focus on continuous improvement of platform reliability and serviceability
• Perform engineering research, case studies, and build prototypes for innovative technologies and their applications at Heroku
• Build product and share knowledge by authoring, maintaining, and/or reviewing code.
• Ensure that we are continuously raising our standard of engineering excellence in AI
• Mentor and guide engineers in technical skills and career advancement
• Participate in the AI team’s on-call rotation to address complex problems in real-time and keep services operational and highly available

Required Skills:

• BS or MS in Engineering, Computer Science, Artificial Intelligence, Machine Learning, or related disciplines or equivalent experience in technical leadership roles.
• Minimum of 5+ years of proven track record in individual-contributor tech leadership roles in software engineering teams, with proven expertise in AI, including generative AI, machine learning, and enterprise / PaaS environments. Experience in multiple AI technologies, services and products, as well as legal and privacy requirements faced by organizations wanting to build AI experiences.
• Experience building, delivering, and operating AI capabilities in AWS infrastructure.
• Experience with machine learning software frameworks including PyTorch, pgvector, JAX, Neuron, MXNet, TensorFlow, or similar technologies.
• Mastery of multiple programming languages and platforms, especially for machine learning use cases.
• Familiarity with hardware optimizations for AI, including Inferentia, Trainium, and GPUs in general.
• 15+ years of software development experience
• Ability to independently craft and deliver large sophisticated projects

• Ability to mentor to team members to support their understanding of the architecture and aid in their technical and professional development
• Experience with agile development methodologies (e.g., scrum)
• Ability to work with other architects, principal architects, engineering managers, and the CTO to craft the architectural vision and roadmap for the organization
• Develops technical performance architecture that solves most complex performance issues
• Ability to drive long-term design strategies that span multiple complex projects, deliver technical reports and presentations to customers and at industry events
• Ability to build an environment for clear and open discussion of all issues. You involve the right people, from your team and others, to resolve critical issues.
• Experience in continuous delivery and short release cycles.

Preferred Skills:

• Familiarity with Heroku and Salesforce platforms

Benefits & Perks

Check out our benefits site which explains our various benefits, including wellbeing reimbursement, generous parental leave, adoption assistance, fertility benefits, and more.

Salesforce Information

Check out Heroku and the Salesforce Engineering Site.

*IN SCHOOL OR GRADUATED WITHIN THE LAST 12 MONTHS? PLEASE VISIT FUTURE FORCE FOR OPPORTUNITIES*

Accommodations

If you require assistance due to a disability applying for open positions please submit a request via this Accommodations Request Form.

Posting Statement

At Salesforce we believe that the business of business is to improve the state of our world. Each of us has a responsibility to drive Equality in our communities and workplaces. We are committed to creating a workforce that reflects society through inclusive programs and initiatives such as equal pay, employee resource groups, inclusive benefits, and more. Learn more about Equality at www.equality.com and explore our company benefits at www.salesforcebenefits.com.

Salesforce is an Equal Employment Opportunity and Affirmative Action Employer. Qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender perception or identity, national origin, age, marital status, protected veteran status, or disability status. Salesforce does not accept unsolicited headhunter and agency resumes. Salesforce will not pay any third-party agency or company that does not have a signed agreement with Salesforce.

Salesforce welcomes all.

Pursuant to the San Francisco Fair Chance Ordinance and the Los Angeles Fair Chance Initiative for Hiring, Salesforce will consider for employment qualified applicants with arrest and conviction records.

For New York-based roles, the base salary hiring range for this position is $213,900 to $355,900.

For Colorado-based roles, the base salary hiring range for this position is $194,400 to $296,600.

For Washington-based roles, the base salary hiring range for this position is $194,400 to $326,300.

For California-based roles, the base salary hiring range for this position is $213,900 to $355,900.