How to infer torch model in scala spark?

We train a model using pytorch, and need to infer it in scala spark tasks with cpu.However, I do not find any resources about it.
torch version :1.7.1

Thank you

It’s unlikely you’ll find spark experts on this forum but you can ask the spark folks here Newest 'apache-spark' Questions - Stack Overflow

And i found this somewhat recent blog post on databricks website PyTorch on Databricks - Introducing the Spark PyTorch Distributor