Studying spin squeezing in Mathematica

I made a simple Mathematica file for simulating spin squeezing numerically and creating density plots of the Husimi Q function, as the one shown below. At the very least it is an interesting exercise on using Mathematica's sparse matrices, sparse matrix exponentiation and multiplication with a vector, and colored density plots in 3D.

squeezing.jpg