# Support Keras 3

**URL:** <https://discuss.ray.io/t/support-keras-3/14792>\
**Category:** RLlib\
**Created:** [May 25, 2024, 3:41pm UTC](https://discuss.ray.io/t/support-keras-3/14792 "2024-05-25T15:41:49Z")\
**Posts on this page:** 2\
**Page:** 1

<div class="post-metadata">

**Author:** ![RocketRider](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/rocketrider/32/6171_2.png) [@RocketRider](https://discuss.ray.io/u/RocketRider)\
**Post date:** [May 25, 2024, 3:41pm UTC](https://discuss.ray.io/t/support-keras-3/14792/1 "2024-05-25T15:41:49Z")

</div>

I am currently working on using Keras 3 with TF2 backend in RLlib.  
But I am facing a few issues.  
To fix the issues I created two pull requests. Could anyone please take a look at these and see if we could merge them?

> <https://github.com/ray-project/ray/pull/44700>
>
> Fixes: https://github.com/ray-project/ray/issues/44671
> @sven1977 could you take… a look?

> <https://github.com/ray-project/ray/pull/45562>
>
> Fix ValueError: Argument \`learning\_rate\` should be float, or an instance of Lear…ningRateSchedule, or a callable (that takes in the current iteration value and returns the corresponding learning rate value). Received instead: learning\_rate=\<tf.Variable 'lr:0' shape=() dtype=float32, numpy=0.0005\>
> 
> Fixes: https://github.com/ray-project/ray/issues/45050

I would really appretierte it if you could look into supporting Keras 3 with different backends.  
Currtenlty I can use Keras 3 with legacy mode, but not with the new version.

---

<div class="post-metadata">

**Author:** ![RocketRider](https://sea2.discourse-cdn.com/flex020/user_avatar/discuss.ray.io/rocketrider/32/6171_2.png) [@RocketRider](https://discuss.ray.io/u/RocketRider)\
**Post date:** [May 28, 2024, 5:27am UTC](https://discuss.ray.io/t/support-keras-3/14792/2 "2024-05-28T05:27:47Z")

</div>

@arturn could you help me again to get these changes merged?