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 WorkThe 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.
We're hiring two Senior Product Designers to shape how developers and AI teams build, train, and run AI on Lightning. You'll own end-to-end product experiences across our platform. That ranges from the first five minutes a solo developer spends with us to the workflows enterprise teams rely on to run models on large-scale compute. It also includes new products like GraphN, which helps any software engineer build AI workflows with no ML background required.
We want designers who think like product owners, learn technical domains fast, and hold a high bar for craft. You'll spend real time with users: running tests, watching sessions, and turning what you learn into better product decisions. You'll use AI the way our customers do, to explore more ideas, prototype in code, and ship faster.
You'll join the Design team and report to our Chief Design Officer. This is a hybrid role based in our San Francisco, Seattle, or New York City office, in office 2 days per week.
What You’ll Do- Own end-to-end experiences across the platform, from self-serve onboarding and pricing to enterprise workflows for training and inference.
- Partner with product and engineering as a peer in deciding what we build and why, not only how it looks.
- Test with users continuously: recruit participants, run sessions, synthesize what you learn, and bring it back into decisions.
- Prototype at the right fidelity, including working prototypes built with AI coding tools, so ideas get tested with users and engineers early.
- Engage with the open source AI community and enterprise AI teams to understand their real workflows and pain points.
- Find and remove friction and papercuts so users succeed from day one.
- Contribute to our design system and help shape how the team uses AI in its own practice.
- 5+ years designing B2B products, ideally developer tools, AI/ML, cloud, or technical SaaS.
- A portfolio that shows product thinking (problem framing, tradeoffs, and outcomes), not only final screens.
- A high level of craft across interaction and visual design.
- A track record of testing with users regularly and changing direction based on what you learned.
- Fluency using AI in your design practice, such as prototyping with Claude Code, Cursor, v0, or Figma Make, and curiosity about where it's heading.
- You learn fast and get energized by technical domains. You don't need an ML background, but you want to understand how developers and AI teams work.
- Comfort with ambiguity and with products at different stages, from early concept to mature features.
- Clear communication, with the ability to explain design rationale and influence cross-functional partners.
- Nice to have: front-end coding experience, and experience with both product-led and sales-led products.
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.
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.
- 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.
Skills Required
- 5+ years of experience designing B2C and B2B software or technical platforms
- Strong portfolio demonstrating end-to-end product thinking across UX, interaction, and visual design
- Experience collaborating with product managers, engineers, and researchers
- Experience iterating quickly and shipping software thoughtfully
- Curiosity and initiative engaging with the open source AI ecosystem
- Ability to work across early-stage prototypes and production features
- Proficiency in Figma and modern design tools
- Excellent communication skills and ability to present design rationale clearly
- Familiarity with machine learning, APIs, or cloud infrastructure
- Front-end coding experience
What We Do
Our platform provides intuitive open source tools, powerful cloud infrastructure and expertise to help you build AI securely.
Why Work With Us
Our team has shaped groundbreaking AI projects like PyTorch and PyTorch Lightning. We’re educators, innovators, and collaborators, united by a mission to democratize AI. A hybrid based company headquartered in New York City, we value in-person time for collaboration while still allowing flexibility.
Gallery
.png){kind=logo}
