Software Engineering Radio - the podcast for professional software developers cover image

Episode 549: William Falcon Optimizing Deep Learning Models

Software Engineering Radio - the podcast for professional software developers

00:00

Using Lightning to Train Machine Learning Models

The traditional PyTorch lightning module trainer integrates with any experiment manager that's out there. So you can find two models from any library you want, open AI and so on. It's like three lines of code to find 200 train these things, torch vision, et cetera. And then as I mentioned, now we've evolved the project that's called lightning now. With the lightning work and the lightning flow, you can actually integrate any framework, even like TensorFlow. We have all the accelerators, so GPUs, HPU, TPUs. You can do jacks. As long as it's machine learning, you can do numpy, sklearn, all

Transcript
Play full episode

The AI-powered Podcast Player

Save insights by tapping your headphones, chat with episodes, discover the best highlights - and more!
App store bannerPlay store banner
Get the app