Keras Core: Keras for TensorFlow, Jax, and PyTorch
1–10 of 73 posts
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#2"We're excited to share with you a new library called Keras Core, a preview version of the future of Keras. In Fall 2023, this library will become Keras 3.0. Keras Core is a full rewrite of the Keras codebase that rebases it on top of a modular backend architecture. It makes it possible to run Keras workflows on top of arbitrary frameworks — starting with TensorFlow, JAX, and PyTorch."
Excited about this one. Please let us know if you have any questions.
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#3Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#4Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#5Keras was already that some years ago. It supported tensorflow, theano, mxnet if my memory is right. And then they ditched everything for tensorflow. At the time it was really hard to use keras without calling backend directly for lots of optimisation, unsupported feature on they API etc... This make the use of Keras not agnostic at all. What's different now ?
PyTorch adoption: back when Keras went hard into TensorFlow in 2018, both TF and PyTorch adoption were about the same with TF having a bit more popularity. Now, most of the papers and models released are PyTorch-first.
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#6Also, are there any examples using this for the coral TPU?
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#7Does this mean the weights output can be backend agnostic? Also, are there any examples using this for the coral TPU?
Coral TPU could be used with Keras Core, but via the TensorFlow backend only.
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#8I worked on the project, happy to answer any questions!
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#9I worked on the project, happy to answer any questions!
Fast forward to now, and my biggest pain point is that all the new models are released on PyTorch, but the PyTorch serving story is still far behind TF Serving. Can this help convert a PyTorch model into a servable SavedModel?
Re: Keras Core: Keras for TensorFlow, Jax, and PyTorch
#10I worked on the project, happy to answer any questions!