New AI Tool Thunder Hopes to Accelerate AI Development

ODSC - Open Data Science
3 min readApr 1, 2024

Thunder, a new compiler designed to turbocharge the training process for deep learning models within the PyTorch ecosystem, hopes to accelerate the development of AI. PyTorch, celebrated for its flexibility and intuitive handling, has become a staple among AI developers.

Get your ODSC East 2024 pass today!

In-Person and Virtual Conference

April 23rd to 25th, 2024

Join us for a deep dive into the latest data science and AI trends, tools, and techniques, from LLMs to data analytics and from machine learning to responsible AI.

REGISTER NOW

Yet, the ever-increasing demand for faster computation and more efficient use of hardware resources calls for innovative solutions. Thunder rises to this challenge by optimizing model execution, thereby facilitating quicker advancements in AI.

What makes Thunder different is its ability to seamlessly integrate with PyTorch’s existing optimization tools. This synergy allows for unprecedented speed improvements, with some training tasks for LLMs, such as those involving 7-billion parameters, witnessing a 40% reduction in processing time compared to standard PyTorch operations.

That’s some pretty big numbers. But the best part is that this enhancement is not confined to single-GPU environments but extends to more complex multi-GPU and distributed training frameworks, leveraging techniques like distributed data-parallel (DDP) and fully sharded data parallel (FSDP).

Another interesting part of Thunder is the fact that its developers spent significant time ensuring that it had a user-friendly design. The compiler enables developers to boost their PyTorch models’ performance with minimal adjustments.

To do this, simplify and incorporate the Thunder.Jit() function, users can activate the compiler’s optimizations and enjoy a more streamlined and efficient training process. For team leads who are looking to supercharge their development lifecycle, this is a significant leap forward.

As you can imagine, its ability to accelerate the training of large language models not only saves valuable time and computational resources but also opens up new avenues for innovation and exploration in the field of artificial intelligence.

Only time will tell how much traction within the AI community Thunder will see. But as it continues to evolve in development, its evolving capabilities are expected to further refine and enhance the efficiency of AI model development.

2024 Data Engineering Summit tickets available now!

In-Person Data Engineering Conference

April 23rd to 24th, 2024 — Boston, MA

At our second annual Data Engineering Summit, Ai+ and ODSC are partnering to bring together the leading experts in data engineering and thousands of practitioners to explore different strategies for making data actionable.

REGISTER NOW

Originally posted on OpenDataScience.com

Read more data science articles on OpenDataScience.com, including tutorials and guides from beginner to advanced levels! Subscribe to our weekly newsletter here and receive the latest news every Thursday. You can also get data science training on-demand wherever you are with our Ai+ Training platform. Interested in attending an ODSC event? Learn more about our upcoming events here.

--

--

ODSC - Open Data Science

Our passion is bringing thousands of the best and brightest data scientists together under one roof for an incredible learning and networking experience.