[Feature Request] Batch inference/representations

### Summary

Having batch inference/representations would allow faster compute for any large scale project

### Detailed Description

Hello!
I am working on a large scale molecule project and have been using Uni-Mol representations, I am implementing a larger batch size for the representation from tools, but I believe you know the architecture better than me, is this something you are working on/ think of implementing in the near future?
Thank you!

### Further Information, Files, and Links

_No response_