[BUG] mlx.JVP returns all boolean answers while mlx.VJP returns the correct floats

[kyrollosyanny]

Describe the bug
I have a function that Im calculating the jacobian of. It works well in mlx.VJP. However, when using JVP, I first get this error [negative] Not supported for bool, use logical_not instead.. Then running it again, the error disappears and I get an all boolean output for the jacobian.

I assume this could be because some operations are not implemented for the forward pass. I am just not sure which ones.

• Version [0.30.4]

[awni]

Could you share some code which reproduces the issue you are seeing?