UniTB framework for universal equivariant Hamiltonian learning | HACKOBAR_