Senior Product Manager, Inference

Who We Are

Lightning AI is the company behind PyTorch Lightning. Founded in 2019, we build an end-to-end platform for developing, training, and deploying AI systems—designed to take ideas from research to production with less friction.

Through our merger with Voltage Park, a neocloud and AI Factory, Lightning AI combines developer-first software with cost-efficient, large-scale compute. Teams get the tools they need for experimentation, training, and production inference, with security, observability, and control built in.

We serve solo researchers, startups, and large enterprises. Lightning AI operates globally with offices in New York City, San Francisco, Seattle, and London, and is backed by Coatue, Index Ventures, Bain Capital Ventures, and Firstminute.

Our Values

Move Fast: We act with speed and precision, breaking down big challenges into achievable steps.

Focus: We complete one goal at a time with care, collaborating as a team to deliver features with precision.

Balance: Sustained performance comes from rest and recovery. We ensure a healthy work-life balance to keep you at your best.

Craftsmanship: Innovation through excellence. Every detail matters, and we take pride in mastering our craft.

Minimal: Simplicity drives our innovation. We eliminate complexity through discipline and focus on what truly matters.

What We're Looking For

We’re looking for a Product Manager focused on Growth who can drive Lightning’s self-serve adoption across thousands of new developers each week. This is a unique opportunity to blend consumer product thinking, growth experimentation, and deep empathy for technical users into a single role.

You’ll own the end-to-end journey of helping developers discover, try, and activate on Lightning AI. From optimizing sign-up flows and onboarding, to scaling A/B tests, to designing delightfully fast feedback loops you’ll work across product, engineering, design, and data to make Lightning feel magical from the first click.

You will be joining the Product Team and report to our Senior Director of Product. This is a hybrid role based in our New York City or San Francisco office with in-office requirements of 2 days per week. The salary range for this role is $160,000 - $210,000.

What You’ll Do

Identify and execute growth levers across activation, retention, and referral

Lead experiments across onboarding, sign-up flows, pricing pages, and messaging

Deep empathy for technical users and an understanding of developer needs, docs, APIs, and integration pain points

Collaborate with design and engineering to rapidly test hypotheses

Use data to uncover friction in the developer journey and remove it

Partner with marketing and community teams to drive traffic, increase activation, and build viral loops via docs, tutorials, and forums

Track and move metrics like DAU/MAU, Time to value, Activation %, PQLs, and expansion revenue

Build with speed learning from failure, doubling down on what works

What You’ll Need

5+ years of product management experience in high-growth, fast-paced environments, 2+ years in consumer/developer products with a trial to paid motion

Bias for action and shipping, paired with the ability to think strategically

Proven track record running A/B tests and using data to drive decisions

Strong analytical skills; comfortable with product instrumentation, metrics, and dashboards

A sincere curiosity for the AI/ML space that enable you to go deep into the technical subject matter in order to better understand our customers and their challenges

Excellent product judgment and a feel for what makes UX “click”

Demonstrates a strong willingness to learn, adapt, and improve by actively seeking and applying feedback from peers, leaders, and customers. Able to adjust approaches based on constructive criticism to drive personal and team growth.

Comfortable with technical concepts and able to work closely with engineering teams to define product specifications and address technical challenges. Direct experience with model training and inference is a plus.

Have a Bachelor's degree in Computer Science, Engineering, or related field

Bonus: Experience with growth loops, referrals, pricing experiments, or viral acquisition

We are committed to offering competitive compensation that reflects the value each team member brings to our mission. Final offers are based on factors such as experience, skills, geographic location, and role expectations. In addition to base salary, our total rewards package for eligible roles includes a discretionary bonus, a meaningful equity component, and comprehensive benefits.

The anticipated annual base salary range for this role is:

$160,000—$275,000 USD

Benefits and Perks

We offer a comprehensive and competitive benefits package designed to support our employees’ health, well-being, and long-term success. Benefits may vary by location, team, and role.

Benefits include:

Comprehensive medical, dental and vision coverage (U.S.); Private medical and dental insurance (U.K.)

Retirement and financial wellness support (U.S.); Pension contribution (U.K.)

Generous paid time off, plus holidays

Paid parental leave

Professional development support

Wellness and work-from-home stipends

Flexible work environment

At Lightning AI, we are committed to fostering an inclusive and diverse workplace. We believe that diverse teams drive innovation and create better products. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity, national origin, age, disability, veteran status, or any other protected characteristic. We are dedicated to building a culture where everyone can thrive and contribute to their fullest potential.