Differentiating non differentiable

[jakubMitura14]

Hello I am in early stages of the model that has non differentiable part (I want to create the graph - discrete structure - from the output of the CNN - hence gradient can back flow from GNN to CNN) . I know that there are some methods of combinatorial optimazation that may help in making the non differentiable problem differentiable. Example of such tool is InferOpt.jl - Hovewer it is not compatible with Jax. I there anything in jax that helps in such usecases

[krzysztofrusek]

Hi, take a look at #12460

[jakubMitura14]

Thanks! I will analyze it