Refolk
Open nowEngineeringProduct

Senior Product Manager, AI Infrastructure

Lightning AI · New York, New York

Location
New York, New York, United States
Level
Senior
Posted
14 October

About this role

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.

The Way We Work

The people who thrive here are builders who move fast, communicate openly, take ownership, and continuously improve themselves, their teams, and our company. Here's what that looks like in practice:

  • Move with Urgency: We move quickly, make thoughtful decisions, and keep momentum. We value action over perfection and learn by shipping.
  • Take Ownership: We own outcomes, not just our individual work. We make decisions that move the company forward and follow through.
  • Communicate Openly: We communicate directly, seek to understand, and create clarity for others. Honest conversations help us move faster together.
  • Build Great Teams: We lead by example, empower others, and create healthy teams where people can do their best work.
  • Raise the Bar: We're always improving ourselves. We learn from feedback, consistently challenge ourselves to grow, and focus on the work that matters most.
  • Think Long-Term: We design for what's next. We create scalable systems, simplify complexity, and use AI and automation to amplify our impact.

What We're Looking For

We’re looking for a Senior Product Manager, Enterprise who thrives at the intersection of customer empathy, technical depth, and execution. You’ll own identifying high-impact opportunities from our enterprise customers and ensure that when something is broken, it gets fixed - fast and well.

This is not just about roadmap delivery: you’ll be the voice of our enterprise customers inside Lightning, ensuring their needs influence our strategy while driving execution with engineering, design, and GTM teams.

You’ll report to our Senior Director of Product, Neil Bhatt. This hybrid role is based in NYC or SF. The salary range for this role is $160,000 - $210,000.

What You’ll Do

  • Identify & Prioritize Opportunities: Partner with enterprise customers to uncover high-revenue, high-value opportunities that shape our product strategy.
  • Own Customer Outcomes: Act as an extension of post-sales ensuring blockers are removed, integrations work, and enterprise adoption expands.
  • Drive Operational Excellence: Build trusted relationships with technical users, ensuring issues are surfaced, owned, and resolved with speed and quality.
  • Bridge Customer & Engineering: Translate developer needs (APIs, integrations, docs, workflows) into intuitive user flows and actionable specs.
  • Partner with Sales & CS: Collaborate on land-and-expand motions, ensuring enterprise success turns into expansion.
  • Execute Relentlessly: Balance strategic thinking with a bias to action driving multiple initiatives forward with clarity, speed, and proactive communication.
  • Build products that developers love: design intuitive user flows and coach an engineering team to ship higher quality products faster.

What You’ll Need

  • 5+ years in product management in high-growth SaaS, including 2+ years in enterprise B2B with land-and-expand or post-sales visibility.
  • Track record of identifying and executing on high-revenue enterprise opportunities while influencing product direction.
  • Strong empathy for technical users; familiarity with APIs, developer workflows, and integration challenges.
  • Analytical fluency: comfortable with instrumentation, metrics, dashboards, and making data-driven calls.
  • Excellent product judgment and an eye for elegant, intuitive UX.
  • Strong collaboration skills with a proven ability to partner with Sales, CS, Marketing and Engineering.
  • Comfort with technical concepts and ability to work closely with engineering to define specs and resolve challenges.
  • Highly coachable with a demonstrated ability to maintain an open mind, take feedback onboard, learn from it, and continuously improve.
    Bachelor’s degree in Computer Science, Engineering, or related field.
  • Bonus: Experience in AI cloud platforms, model training, or inference.

Benefits and Perks

We offer a comprehensive and competitive benefits package designed to support our employees’ health, well-being, and long-term success:

  • Comprehensive Health Coverage: Medical, dental, and vision coverage for employees and eligible dependents.
  • Meaningful Equity: RSUs that give employees a stake in the company's long-term success.
  • Retirement Savings: 401(k) matching (U.S.) and pension contributions (U.K.).
  • Flexible Time Off: Unlimited PTO, company holidays, and floating holidays to support work-life balance.
  • Company-Wide Winter Break: Two weeks of company closure each winter to disconnect and recharge.
  • Paid Parental & Family Leave: Paid leave to support you and your family through life's important moments.
  • Professional Development: Annual learning and development allowance to support your professional growth.
  • Wellness Benefits: Wellness and work-from-home stipends to support your physical and mental well-being.
  • Sabbatical Program: Four weeks of paid sabbatical leave after four years of service.
  • Flexible Work: Flexible schedules and a hybrid work model for our office-based teams.
  • In-Office Meals: Complimentary meals at our office hubs.

Benefits may vary by location, team, and role.

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.

As published by Lightning AI. Applications are handled on their site.

Skills this posting mentions

Artificial IntelligenceData DrivenSoftware as a Service

About Lightning AI

The AI development platform - From idea to AI, Lightning fast ⚡️. Code together. Prototype. Train on GPUs. Scale. Serve. From your browser - with zero setup. AI Studio is your laptop on the cloud. Zero setup. Always ready. Persistent storage and environments. Code on CPU. Debug on GPU. Scale to multi-node. Run sweeps, jobs and more. Scale models with PyTorch Lightning, Fabric, Lit-GPT, torchmetrics and more.

All 49 openings at Lightning AI

One click, then it is written

Apply to Lightning AI with a resume written for this role.

Queue Senior Product Manager, AI Infrastructure and I read the posting, rewrite your resume against it, draft the cover letter, and score the fit. Then you press send, or press one button and I fill in Lightning AI’s form for you.

  1. 01Drop your resume

    A PDF or a LinkedIn URL. About a minute, once.

  2. 02I rank the openings

    Every weekday morning, the live catalog scored against your history. Up to 20 worth your time, not two hundred links.

  3. 03Each one is written up

    Resume rewritten for the posting, a cover letter, a fit score. Press send, or let me fill in the form.

  • 25 sent a week, free
  • No card
  • Nothing sent until you say so

More roles at Lightning AI

See all

Similar roles elsewhere

See more

Put this to work

Paste your career in once. Every application after that is written for you.

Drop a resume or a LinkedIn URL. I rank the live openings against it, rewrite the resume and write a cover letter for the best of them, and fill in the employer's form when you press the button. You read, you decide what goes out.

  1. 01Drop your resume

    A PDF or a LinkedIn URL. About a minute, once.

  2. 02I rank the openings

    Every weekday morning, the live catalog scored against your history. Up to 20 worth your time, not two hundred links.

  3. 03Each one is written up

    Resume rewritten for the posting, a cover letter, a fit score. Press send, or let me fill in the form.

  • New matches ranked and written before you are up.
  • Every bullet stays inside what your history supports. Nothing invented.
  • Queued, submitted, interviewing, offer: one screen, not a spreadsheet.

500 free credits on sign-up. No card. Nothing is sent until you say so.

Listed from the job board Lightning AI publishes. Refolk is not the employer and does not handle their hiring. Applications go to Lightning AI directly.

Hiring at Lightning AI? Post this role to the Refolk talent pool for $299, once. No commission, refunded if I find nobody. Post a role for $299