Introduction to the PyTorch Lightning Ecosystem


This talk introduces the PyTorch lightning ecosystem one step at a time. We start with Lightning Flash, an AI factory and an easy-to-use framework for getting started with the Lightning ecosystem. With Flash, we showcase a basic NLP task that classifies the toxicity level of user comments. Then, we delve into the core components of the PyTorch Lightning framework. Using these components, we show how to implement state-of-the-art research without worrying about the hassles of engineering.

Session Outline

Part 1: Predefined tasks with Lightning Flash

We showcase a Natural Language Processing (NLP) model for detecting toxicity in plain text with Lightning Flash. We will benefit from utilizing a clean API that allows us to simply plug in any state-of-the-art model from the transformers repository.

Part 2: Intro to PyTorch Lightning core components with Computer Vision

We start with a brief introduction of Pytorch Lightning’s core building blocks and explore how they fit into the typical research/data scientist development flow. We demonstrate the usage of these building blocks while building a standard computer vision task - multi-label classification.

Part 3: Using PyTorch Lightning at Scale

We will then showcase how you can easily scale your training for a large dataset over various accelerators such as GPUs and TPUs. We'll also go through the basic API internals of how PyTorch Lightning succeeds in abstracting the accelerator logic from users with support for distributed strategies, allowing them to focus on writing accelerator-agnostic code.

Background Knowledge

Some familiarity with Python, deep learning terminology, and the basics of neural networks.


Jirika is working in Machine learning and Data science for several years. He has done Ph.D. in Medical Imaging. In parallel, he gains practical experience while he has been working for a few IT companies as a consultant or data scientist. Actually,he is focusing on exploring interesting world problems and solving them with state-of-the-art techniques.
He has developed several open-source python packages, He is the core contributor of `PyTorch-Lightning` and `TorchMetrics` and actively participating in other well-known projects.

Open Data Science




Open Data Science
One Broadway
Cambridge, MA 02142

Privacy Settings
We use cookies to enhance your experience while using our website. If you are using our Services via a browser you can restrict, block or remove cookies through your web browser settings. We also use content and scripts from third parties that may use tracking technologies. You can selectively provide your consent below to allow such third party embeds. For complete information about the cookies we use, data we collect and how we process them, please check our Privacy Policy
Consent to display content from Youtube
Consent to display content from Vimeo
Google Maps
Consent to display content from Google