A statistical mechanics approach to mixing in stratified fluids