Maybe, but a specific example showing how to do it would have been a more compelling argument.
(castform founder here) we should have made it more prominent on the blogpost but here's the github repo: https://github.com/castform-ai/benchmax/tree/main/examples/n...