Great! Thx for the new link. I will go through this example thoroughly and apply the diffrax to my code.
Amazing guess, it is a geochemical model! I am new to PyMC and I never expect to meet you, a core developer, who are so familiar with the geochemical theory haha!