We should support this in the PyO3 integration, but let's leave it for a future PR _Originally posted by @Luthaf in https://github.com/metatensor/dlpk/pull/16#discussion_r2737256697_