From the PyTorch Lightning creators. Own your AI, don. Backed by Index.
About the role
We’re looking for a Backend Engineer to help develop and scale the Lightning AI platform, spanning APIs, infrastructure, billing, security, and integrations. In this role, you’ll take ownership of key features and drive them end-to-end, working closely with engineering, product, and design to ship high-quality, iterative improvements. This is an opportunity to build systems that support real-world machine learning workloads at scale.
What they're looking for
- Extensive experience writing efficient, maintainable code in Go. Familiarity with best practices in testing, readability, and optimization is essential
- Broad knowledge of backend and frontend systems, with experience building and scaling APIs, CLI tools, billing systems, and security protocols. Exposure to frontend frameworks is a plus
- Experience with cloud platforms like AWS, GCP, or Azure and designing systems that are reliable, performant, and scalable. Familiarity with containerization (Docker, Kubernetes) is a strong advantage
- Proven ability to take ownership of key features and drive end-to-end development
- Commitment to software quality, test automation, and continuous delivery. Experience proactively managing technical debt and speeding up time to production
- Hands-on experience as a backend or product engineer in a SaaS technology company
More about this role
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 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...
Browse similar: AI jobs · AI startup jobs · Startup jobs · San Francisco Bay Area