Akihiro Nitta's Contact Details

Akihiro Nitta's Company Details

PyTorch Lightning

New York, NY, US • 20 - 49 Employees
Software

Iterate on your research ideas in Lightning Speed!Writing code for complex deep learning research experiments tends to be... complex. Prototyping new research ideas can get painful when training on massive data sets or multi-layered deep networks. It requires nontrivial engineering: huge codebases to maintain and debug, tons of boilerplate to scale to distributed hardware, and creates a lot of room for errors.PyTorch Lightning is a lightweight machine learning framework that handles most of the engineering work, leaving you to focus on the science. It's more of a PyTorch style-guide than a framework, setting a standard on how to structure your deep learning PyTorch code. Lightning ensures that when your network becomes complex your code doesn't.Refactoring your models to lightning is simple, allows you to get rid of a ton of boilerplate, reduce cognitive load, and gives you the ultimate flexibility to iterate on research ideas faster with all the latest deep learning best practices.We're venture funded and backed by some of the top VC funds in the world: Index Ventures, Bain Capital Ventures, and First Minute Capital.Grid.ai is our native platform for training models at scale on the cloud!Sign up for early access to supercharge your research and production work: grid.ai