Keras 3.0 officially released: available for TensorFlow, JAX and PyTorch-AI-php.cn

Home

Technology peripherals

Keras 3.0 officially released: available for TensorFlow, JAX and PyTorch

WBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWB

Nov 29, 2023 pm 03:43 PM

industrykeras 3

After 5 months of update iterations, Keras 3.0 is finally here.

"Big news: We just released version Keras 3.0!" François Chollet, the father of Keras, on X Expressed excitement. "Now you can run Keras on JAX, TensorFlow and PyTorch frameworks..."

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Regarding this update, Keras officials stated that this version took them 5 years to complete. It took several months for the public beta to be completed. Keras 3.0 is a complete rewrite of Keras that allows you to run Keras workflows on top of JAX, TensorFlow, or PyTorch. The new version also has new large model training and deployment capabilities. You can choose the framework that suits you best, or switch from one framework to another based on your current goals.

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Keras address: https://keras.io/keras_3/

has been used by more than 2.5 million Keras, used by developers, welcomes version 3.0

Keras API is available for JAX, TensorFlow and PyTorch. Existing tf.keras models using only built-in layers can be run in JAX and PyTorch!

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Keras 3 works seamlessly with any JAX, TensorFlow and PyTorch workflow. Keras 3 is not only suitable for Keras-centric workflows such as defining Keras models, optimizers, losses and metrics, it is also designed to integrate seamlessly with JAX, TensorFlow and PyTorch low-level backend native workflows when training Keras models. , you can choose to use JAX training, TensorFlow training, PyTorch training, or use it as part of a JAX or PyTorch model. There is no problem with the above operations. Keras 3 provides the same degree of low-level implementation flexibility in JAX and PyTorch that tf.keras does in TensorFlow.

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Pre-trained model. You can now use a variety of pretrained models in Keras 3. There are now 40 Keras application models available in the backend, in addition to a large number of pre-trained models existing in KerasCV and KerasNLP (such as BERT, T5, YOLOv8, Whisper, SAM etc) also applies to all backends.

Keras 3 is highly backwards compatible with Keras 2: Keras 3 now implements the public API interface of Keras 2. Most users can run Keras scripts on Keras 3 without changing any code. If you are not used to using Keras 3, you can choose to ignore the new version update and continue to use Keras 2 with TensorFlow.

Keras 3 supports cross-framework data pipeline for all backends. Multi-frame machine learning also means multi-frame data loading and preprocessing. Keras 3 models can be trained using a variety of data pipelines, whether you are using JAX, PyTorch or the TensorFlow backend:

tf.data. Dataset pipelines.
torch.utils.data.DataLoader object.
NumPy arrays and Pandas dataframes.
The keras.utils.PyDataset object for Keras.

A new distributed API for large-scale data parallelism and model parallelism. This update is currently only available for the JAX backend, with TensorFlow and PyTorch support coming soon.

As for why this change is being rolled out, the Keras team stated that in recent years, as the model size has become larger and larger, they want to shard the multi-device model ( sharding) problem provides Keras solution. The team designed the API to make model definition, training logic, and sharding configuration completely independent, which means that the model can be run as if it were on a single device. You can then add sharding configuration to any model while training the model.

Data parallelism (copying a small model identically on multiple devices) can be handled with just two lines:

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Next is model parallelism. This API allows you to configure the layout of each variable and each output tensor via regular expressions. This makes it easy to quickly specify the same layout for an entire variable category.

Keras 3.0正式发布：可用于TensorFlow、JAX和PyTorch

Finally, the Keras team has collected many questions that everyone is concerned about and answered them. Interested readers can go to the Keras official website to learn more.

The above is the detailed content of Keras 3.0 officially released: available for TensorFlow, JAX and PyTorch. For more information, please follow other related articles on the PHP Chinese website!

Statement

This article is reproduced at:机器之心. If there is any infringement, please contact admin@php.cn delete

You Must Build Workplace AI Behind A Veil Of IgnoranceApr 29, 2025 am 11:15 AM

In John Rawls' seminal 1971 book The Theory of Justice, he proposed a thought experiment that we should take as the core of today's AI design and use decision-making: the veil of ignorance. This philosophy provides a simple tool for understanding equity and also provides a blueprint for leaders to use this understanding to design and implement AI equitably. Imagine that you are making rules for a new society. But there is a premise: you don’t know in advance what role you will play in this society. You may end up being rich or poor, healthy or disabled, belonging to a majority or marginal minority. Operating under this "veil of ignorance" prevents rule makers from making decisions that benefit themselves. On the contrary, people will be more motivated to formulate public

Decisions, Decisions… Next Steps For Practical Applied AIApr 29, 2025 am 11:14 AM

Numerous companies specialize in robotic process automation (RPA), offering bots to automate repetitive tasks—UiPath, Automation Anywhere, Blue Prism, and others. Meanwhile, process mining, orchestration, and intelligent document processing speciali

The Agents Are Coming – More On What We Will Do Next To AI PartnersApr 29, 2025 am 11:13 AM

The future of AI is moving beyond simple word prediction and conversational simulation; AI agents are emerging, capable of independent action and task completion. This shift is already evident in tools like Anthropic's Claude. AI Agents: Research a

Why Empathy Is More Important Than Control For Leaders In An AI-Driven FutureApr 29, 2025 am 11:12 AM

Rapid technological advancements necessitate a forward-looking perspective on the future of work. What happens when AI transcends mere productivity enhancement and begins shaping our societal structures? Topher McDougal's upcoming book, Gaia Wakes:

AI For Product Classification: Can Machines Master Tax Law?Apr 29, 2025 am 11:11 AM

Product classification, often involving complex codes like "HS 8471.30" from systems such as the Harmonized System (HS), is crucial for international trade and domestic sales. These codes ensure correct tax application, impacting every inv

Could Data Center Demand Spark A Climate Tech Rebound?Apr 29, 2025 am 11:10 AM

The future of energy consumption in data centers and climate technology investment This article explores the surge in energy consumption in AI-driven data centers and its impact on climate change, and analyzes innovative solutions and policy recommendations to address this challenge. Challenges of energy demand: Large and ultra-large-scale data centers consume huge power, comparable to the sum of hundreds of thousands of ordinary North American families, and emerging AI ultra-large-scale centers consume dozens of times more power than this. In the first eight months of 2024, Microsoft, Meta, Google and Amazon have invested approximately US$125 billion in the construction and operation of AI data centers (JP Morgan, 2024) (Table 1). Growing energy demand is both a challenge and an opportunity. According to Canary Media, the looming electricity

AI And Hollywood's Next Golden AgeApr 29, 2025 am 11:09 AM

Generative AI is revolutionizing film and television production. Luma's Ray 2 model, as well as Runway's Gen-4, OpenAI's Sora, Google's Veo and other new models, are improving the quality of generated videos at an unprecedented speed. These models can easily create complex special effects and realistic scenes, even short video clips and camera-perceived motion effects have been achieved. While the manipulation and consistency of these tools still need to be improved, the speed of progress is amazing. Generative video is becoming an independent medium. Some models are good at animation production, while others are good at live-action images. It is worth noting that Adobe's Firefly and Moonvalley's Ma

Is ChatGPT Slowly Becoming AI's Biggest Yes-Man?Apr 29, 2025 am 11:08 AM

ChatGPT user experience declines: is it a model degradation or user expectations? Recently, a large number of ChatGPT paid users have complained about their performance degradation, which has attracted widespread attention. Users reported slower responses to models, shorter answers, lack of help, and even more hallucinations. Some users expressed dissatisfaction on social media, pointing out that ChatGPT has become “too flattering” and tends to verify user views rather than provide critical feedback. This not only affects the user experience, but also brings actual losses to corporate customers, such as reduced productivity and waste of computing resources. Evidence of performance degradation Many users have reported significant degradation in ChatGPT performance, especially in older models such as GPT-4 (which will soon be discontinued from service at the end of this month). this

See all articles