Google expands TensorFlow open-source tooling for accelerated machine studying improvement

Be part of high executives in San Francisco on July 11-12, to listen to how leaders are integrating and optimizing AI investments for achievement. Be taught Extra

The massive synthetic intelligence (AI) information at Google I/O at the moment is the launch of the corporate’s PaLM 2 massive language mannequin, however that’s not the one AI information on the occasion.

The corporate can also be rolling out a sequence of open-source machine studying (ML) know-how updates and enhancements for the rising TensorFlow ecosystem. TensorFlow is an open-source know-how effort, led by Google, that gives ML instruments to assist builders construct and prepare fashions.

Google is launching its new DTensor know-how at Google I/O. This know-how brings new parallelism strategies to ML coaching, serving to to enhance mannequin coaching and scaling effectivity.

Google mixed parallelism with DTensor
Picture credit score: Google

There’s additionally a preview launch of the TF Quantization API, which is meant to assist make fashions extra resource-efficient general and thus scale back the price of improvement.


Remodel 2023

Be part of us in San Francisco on July 11-12, the place high executives will share how they’ve built-in and optimized AI investments for achievement and averted frequent pitfalls.


Register Now

A key a part of the TensorFlow ecosystem is the Keras API suite, which offers a set of Python language-based deep studying capabilities on high of the core TensorFlow know-how. Google is saying a pair of latest Keras instruments: KerasCV for pc imaginative and prescient (CV) purposes, and KerasNLP for pure language processing (NLP).

“An enormous a part of what we’re when it comes to the tooling and the open-source area is de facto driving new capabilities and new effectivity and new efficiency,” Alex Spinelli, Google’s vp of product administration for machine studying, advised VentureBeat. “Completely Google will construct superior, wonderful AI and ML into its merchandise, however we additionally need to sort of create a rising tide that lifts all ships, so we’re actually dedicated to our open supply methods, and enabling builders at massive.”

TensorFlow stays the ‘workhouse’ of machine studying at Google

In an period the place massive language fashions (LLMs) are all the fad, Spinelli emphasised that it’s now much more important than ever to have the fitting ML coaching instruments.

“TensorFlow continues to be at the moment the workhorse of machine studying,” he stated. “It’s nonetheless … the elemental underlying infrastructure [in Google] that powers plenty of our personal machine studying developments.”

To that finish, the DTensor updates will present extra “horsepower” as the necessities of ML coaching proceed to develop. DTensor introduces extra parallelization capabilities to assist optimize coaching workflows.

Spinelli stated that ML general is simply getting extra hungry for knowledge and compute sources. As such, discovering methods to enhance efficiency with a purpose to course of extra knowledge to serve the wants of more and more bigger fashions is extraordinarily vital. The brand new Keras updates will present much more energy, with modular elements that truly let builders construct their very own pc imaginative and prescient and pure language processing capabilities. 

Nonetheless extra energy will come to TensorFlow because of the brand new JAX2TF know-how. JAX is a analysis framework for AI, extensively used at Google as a computational library, to construct applied sciences such because the Bard AI chatbot. With JAX2TF, fashions written in JAX will now be extra simply usable with the TensorFlow ecosystem.

“One of many issues that we’re actually enthusiastic about is how this stuff are going to make their means into merchandise — and watch that developer neighborhood flourish,” he stated.

PyTorch vs TensorFlow

Whereas TensorFlow is the workhorse of Google’s ML efforts, it’s not the one open-source ML coaching library.

In recent times the open-source PyTorch framework, initially created by Fb (now Meta), has develop into more and more well-liked. In 2022, Meta contributed PyTorch to the Linux Basis, creating the brand new PyTorch Basis, a multi-stakeholder effort with an open governance mannequin.

Spinelli stated that what Google is attempting to do is help developer alternative in terms of ML tooling. He additionally famous that TensorFlow isn’t simply an ML framework, it’s an entire ecosystem of instruments for ML that may assist help coaching and improvement for a broad vary of use instances and deployment situations.

“This is similar set of applied sciences, basically, that Google makes use of to construct machine studying,” Spinelli stated. “I feel we’ve got a very aggressive providing if you happen to actually need to construct large-scale high-performance methods and also you need to know that these are going to work on all of the infrastructures of the long run.”

One factor Google apparently won’t be doing is following Meta’s lead and creating an unbiased TensorFlor Basis group.

“We really feel fairly comfy with the best way it’s developed at the moment and the best way it’s managed,” Spinelli stated. “We really feel fairly comfy about a few of these nice updates that we’re releasing now.”

VentureBeat’s mission is to be a digital city sq. for technical decision-makers to achieve information about transformative enterprise know-how and transact. Uncover our Briefings.

Related Articles


Please enter your comment!
Please enter your name here

Latest Articles