Support for batch predictions