Back to jobs
New

Staff Software Engineer, Gen AI

Remote - United States

About the Staff Software Engineer, Gen AI at Headspace:

As a Staff Software Engineer, Generative AI at Headspace, you'll play a pivotal role in advancing the company's capabilities in artificial intelligence, particularly in developing, optimizing, and deploying generative AI models. Your work will enable innovative applications of AI that align with Headspace's mission of improving health and happiness worldwide. By developing state-of-the-art frameworks and tools, you will empower engineers, data scientists, and other practitioners to maximize the potential of generative AI across the organization.

What you will do:

  • Generative AI Model Development: Lead the design, training, and optimization of large-scale generative AI models to support innovative use cases across Headspace.
  • End-to-End AI Solutioning: Own the lifecycle of generative AI initiatives, from ideation to deployment, ensuring robust and scalable production systems.
  • Platform Optimization: Build and enhance the generative AI platform using AWS and modern development tools, with a focus on Infrastructure as Code (IaC) practices.
  • Scalability and Reliability: Ensure the scalability and reliability of generative AI systems by implementing robust CI/CD pipelines, monitoring solutions, and alerting systems.
  • Cross-Functional Collaboration: Partner with data scientists, engineers, and product teams to integrate generative AI capabilities into Headspace’s products and services.
  • Knowledge Sharing and Mentorship: Guide and mentor team members, fostering a culture of technical excellence and continuous learning.

What you will bring:

Required Skills:


    • 4+ years of software engineering experience with a focus on AI/ML systems, including developing and deploying generative models.
    • Proficiency in machine learning frameworks (e.g., TensorFlow, PyTorch) and Python.
    • Experience with database management / querying, e.g. SQL, MongoDB, etc.
    • Deep expertise in AWS and IaC tools such as Terraform, with a minimum of 3 years of hands-on experience.
    • Strong understanding of ML infrastructure, including data pipelines, model serving, and scaling.
    • Proven track record of delivering high-impact AI solutions in production environments.
    • Experience implementing and maintaining AI/ML best practices, including CI/CD pipelines and operational monitoring.
    • Skilled at translating complex technical concepts into actionable insights for diverse stakeholders.
    • A collaborative mindset with the ability to mentor and uplift team capabilities.
    • Bachelor’s or advanced degree in Computer Science, Engineering, or a related technical field.

Pay & Benefits:

The base salary range for this role is determined by a number of factors, including but not limited to skills and scope required, relevant licensure and certifications, and unique relevant experience and job-related skills. The base salary range for this role is $131,414 - $197,100.

At Headspace, cash salary is but one component of our Total Rewards package. We’re proud of our robust package inclusive of: base salary, stock awards, comprehensive healthcare coverage, monthly wellness stipend, retirement savings match, lifetime Headspace membership, unlimited, free mental health coaching, generous parental leave, and much more. Paid performance incentives are also included for those in eligible roles. Additional details about our Total Rewards package will be provided during the recruitment process.

How we feel about Diversity, Equity, Inclusion and Belonging:

Headspace is committed to bringing together humans from different backgrounds and perspectives, providing employees with a safe and welcoming work environment free of discrimination and harassment. We strive to create a diverse & inclusive environment where everyone can thrive, feel a sense of belonging, and do impactful work together. 

As an equal opportunity employer, we prohibit any unlawful discrimination against a job applicant on the basis of their race, color, religion, gender, gender identity, gender expression, sexual orientation, national origin, family or parental status, disability*, age, veteran status, or any other status protected by the laws or regulations in the locations where we operate. We respect the laws enforced by the EEOC and are dedicated to going above and beyond in fostering diversity across our workplace. 

*Applicants with disabilities may be entitled to reasonable accommodation under the terms of the Americans with Disabilities Act and certain state or local laws. A reasonable accommodation is a change in the way things are normally done which will ensure an equal employment opportunity without imposing undue hardship on Headspace. Please inform our Talent team by filling out this form if you need any assistance completing any forms or to otherwise participate in the application or interview process.

Headspace participates in the E-Verify Program.

Privacy Statement

All member records are protected according to our Privacy Policy. Further, while employees of Headspace (formerly Ginger) cannot access Headspace products/services, they will be offered benefits according to the company's benefit plan. To ensure we are adhering to best practice and ethical guidelines in the field of mental health, we take care to avoid dual relationships. A dual relationship occurs when a mental health care provider has a second, significantly different relationship with their client in addition to the traditional client-therapist relationship—including, for example, a managerial relationship.  

As such, Headspace requests that individuals who have received coaching or clinical services at Headspace wait until their care with Headspace is complete before applying for a position. If someone with a Headspace account is hired for a position, please note their account will be deactivated and they will not be able to use Headspace services for the duration of their employment. 

Further, if Headspace cannot find a role that fails to resolve an ethical issue associated with a dual relationship, Headspace may need to take steps to ensure ethical obligations are being adhered to, including a delayed start date or a potential leave of absence. Such steps would be taken to protect both the former member, as well as any relevant individuals from their care team, from impairment, risk of exploitation, or harm.

For how how we will use the personal information you provide as part of the application process, please see: https://www.headspace.com/applicant-notice

Create a Job Alert

Interested in building your career at Headspace? Get future opportunities sent straight to your email.

Create alert

Apply for this job

*

indicates a required field

Resume/CV*

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter

Accepted file types: pdf, doc, docx, txt, rtf

Employment

Select...
Select...
Select...
Select...
Select...

US Custom Demographic Survey

In our company-wide effort to continue to prioritize diversity and inclusion, we invite candidates to respond to both the EEOC Voluntary Self-Identification survey as well as a custom self-identification survey as a part of the application process. The EEOC survey helps us remain compliant with government reporting requirements (full information and instructions below). We have also created custom self-identification demographic questions to expand upon specific categories, offering more inclusive options that more accurately reflect the diverse global workforce. While this survey is a required section of our application, your level of disclosure is completely voluntary. Participating also includes selecting “I don’t wish to answer” as your choice for any of the categories. This data helps us to better understand how our DEIB efforts are tracking across various measures. Aggregated personal data will be reported on and shared with the necessary stakeholders as needed.  This data will be scrubbed of all possible identifying information, is maintained in a confidential file and will not be considered in the hiring process. 

Select...
Select...
Select...

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Headspace’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Select...
Select...
Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Select...

Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Select...

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.