Comment by datanecdote

It is true that Jax cannot differentiate through C code. But it can differentiate through python code that was written to accept Numpy.