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