How to find jacobian of neural network model for current variables only and not history variables

My data for training neural network model contains current variables and also history variables to find best fit for non-linear function. Training perfomance is very good and model can predict data with very low error.

My main goal is to find tangent for the learnt model. I am using
f(array of input parameters both current and history variables) = NN_model
Zygote.jacobian(f,[ array of input parameters both current and history variables])
function to find tangent. This does not give results as expected. I want to find jacobian with respect to current variables only.
Zygote.jacobian(f,[ array of only parameters current]) → this gives error
Would it be possible with Zygote package or any other recommendations please?