Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

custom_gradient not working with JAX backend #21105

Open
larschristensen opened this issue Mar 28, 2025 · 0 comments
Open

custom_gradient not working with JAX backend #21105

larschristensen opened this issue Mar 28, 2025 · 0 comments
Assignees
@sachinprasadhs
Labels
backend:jax type:Bug

Comments

@larschristensen
Copy link

When trying to use the custom_gradient decorator with a JAX backend as shown in the example located here
custom_gradient_example, I get an error message saying TypeError: 'NoneType' object is not callable.

It works without a problem on the TensorFlow backend, but not JAX. The behaviour is seen both on keras v3.9.0 and on keras-nightly.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@sachinprasadhs sachinprasadhs

Labels
backend:jax type:Bug
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@larschristensen @sachinprasadhs @sonali-kumari1