In Tattvartha Sutra, Chapter 4 is the one that discusses heavenly beings (devas) and naraka beings in the context of gati / cosmology.

So if you mean Deva lok and the beings there, look at Chapter 4. If you want, I can also tell you the exact sutras in that chapter related to devas, their classes, and their dwellings.