Session Crashed for an unknown reason

[Sumya-Oishe]

Describe the current behavior
The same code that was running fine in the Colab free version is now causing my Colab Pro session to crash. During execution, it abruptly terminates with the message: "Your session crashed for an unknown reason." A screenshot of warnings are attached below.

Describe the expected behavior
The code should run in Colab Pro as it did in the free version, without crashes or interruptions. I expected to have improved performance and stability due to the increased resources available in Colab Pro.

What web browser you are using
Google Chrome.

Additional context
I am using the ceviche and autograd libraries in my code. Here is a link to a minimal, public, self-contained notebook that reproduces the issue: https://github.com/Sumya-Oishe/Adjoint-Opt/blob/210de35bf4204f769df38b5b841a71cd4649bd09/Classify_0_1_2_(ceviche).ipynb