Open Craigacp opened 2 years ago
Thanks for the tip! I'm looking forward to taking a close look. By the way, do you have a MuZero implementation on Tribuo? ... I am asking because I want to export models trained in DJL from a small MuZero sample implementation to run in a browser using ONNX Runtime Web.
We don't have a MuZero implementation, Tribuo's focus is more on being scitkit-learn in Java than being a general purpose deep learning system. We do classification & regression rather than reinforcement learning, and haven't spent too much time thinking about what an RL system would look like in Java.
Thanx for the insight to Tribuo.
I came across this repo from the DJL repo, and you might be interested in the ONNX export functionality we built in Tribuo's last release (v4.2.0). We have a separate module (which only depends on protobuf) that makes writing ONNX models from Java significantly easier, it provides similar helper functionality to the Python ONNX module. We don't have full coverage of the space of ONNX operators yet, but we'd be happy to work with you to expand that set.
The Maven artifact is here, the javadoc is here, and our repo is on GitHub.
You can see examples of how we build ONNX models in the rest of Tribuo (e.g. exporting a FactorizationMachine).