HenryPowell:

produces output until the size of the second dimension of x reaches 14 at which point the core dumped error message occurs.


I’ve hit a similar issue in the past and were seeing that a vectorized code path was used internally starting from a specific length of the tensor. Could you check the backtrace via:
gdb --args python script.py
...
run
...
bt