I am using fmod or remainder to compute a sawtooth waveform, where the divisor argument ‘other’ is computed by some layers (in this case it is for defining the sawtooth frequency), thus it requires gradient.
When performing backpropagation it gives the following error for fmod or remainder:

RuntimeError: the derivative for ‘other’ is not implemented

Is there any workaround please ?

The input argument is a torch.arange(size) which is then shaped as a saw thanks to the modulo operation.

x = torch.arange has a constant derivative/slope, however through the modulo, it becomes non defined at the modulo jump, which makes it non differentiable in each of these points. So as it is, indeed there is an issue for gradients I guess.

I alternatively tried to generate sawtooth as wavetables, by using interpolate to rescale a given sawtooth at a reference frequency to any other frequency, feeding the ratio of the wavetable reference frequency and the predicted f0.

However, it seems as well that interpolate does not accept a gradient variable in the scale factor. Is it correct ?

In this case it seems as well that backpropagation is not possible through the interpolate scale factor.

I am not very sure regarding the use case but you can try this function. It seems to work as a very good approximation and everything seems differentiable. I have just made some back of the envelope calculations regarding how gradient might work in such a situation. Hope you find it useful.

fmod( x, a ) ≈ ( a / π ) arctan( tan( π ( x / a - 0.5 ) ) ) + a / 2