Standardise export approach across code base

The following discussion from !1385 (merged) should be addressed:

• @HongtaoYang started a discussion: (+2 comments)

  If we use __all__, then we don't need re-exports (ruff doc). We should choose one method and be consistent across the codebase.

  For now I'm going to go with re-export and delete the __all__