It’s a bit hard to narrow down the issue from the stacktrace, but you should check if

you are explicitly using inplace operations and replace them with their out-of-place equivalents
you are trying to use stale forward activations as described in this post 5