Adjust code for multiple shapes

examples/dem.py

examples/lj.py

src/pairs/ir/lit.py

src/pairs/ir/symbols.py

src/pairs/mapping/funcs.py

src/pairs/sim/interaction.py

src/pairs/sim/neighbor_lists.py

src/pairs/sim/simulation.py

src/pairs/transformations/expressions.py