Tribuo VS server

It seems to have been picked up by Eclipse and there is also Oracle Labs' Tribuo and Deep Java Library. All seem active, but I don't know much about any of them. I agree it's probably best to follow the community and use a more popular tool like PyTorch.

We built model & data provenance into our open source ML library, though it's admittedly not the W3C PROV standard. There were a few gaps in it until we built an automated reproducibility system on top of it, but now it's pretty solid for all the algorithms we implement. Unfortunately some of the things we wrap (notably TensorFlow) aren't reproducible enough due to some unfixed bugs. There's an overview of the provenance system in this reprise of the JavaOne talk I gave here https://www.youtube.com/watch?v=GXOMjq2OS_c. The library is on GitHub - https://github.com/oracle/tribuo.

Although it may be a bit more work, you can do both machine learning and AI in Java. If you are doing deep learning, you can use DeepJavaLibrary (I do work on this one at Amazon). If you are looking for other ML algorithms, I have seen Smile, Tribuo, or some around Spark.

We've been developing Tribuo on Github for two years now, MS are very actively developing ONNX Runtime (and the Java layer is fairly thin and wrapped over the same C API they use for node.js and C#), and things like XGBoost and LibSVM have been around for many years and the Java bits are developed in tree with the rest of the code so updated along with it. Amazon have a team of people working on DJL, though you'd have to ask them what their plans are.

FWIW, Oracle actually did release a Java ML library - https://github.com/oracle/tribuo.

Tribuo (tribuo.org, github.com/oracle/tribuo). ONNX export support is there for 2 models at the moment in main, there's a PR for factorization machines which supports ONNX export, and we plan to add another couple of models and maybe ensembles before the upcoming release. Plus I need to write a tutorial on how it all works, but you can check the tests in the meantime.

You can do ML & data science in Java (full disclosure: I help run TensorFlow-Java, I maintain ONNX Runtime's Java interface, and I'm the lead developer on Oracle Labs' Java ML library Tribuo, so I'm pretty biased). It tends not to be as favoured in research, though I've published academic ML papers which used Java implementations. People do deploy ML models quite a bit in Java in industry.

It might be worth having a look at the ONNX Runtime Java API in addition to TF-Java, it'll let you deploy the rest of the HuggingFace pytorch models that don't have TF equivalents. I built the Java API a few years ago, and it's now a supported part of the ONNX Runtime project. We use it in Tribuo to provide one of our text feature embedding classes (BERTFeatureExtractor).

The IJava notebook kernel works pretty well for data science on top of Java. We use it in Tribuo to write all our tutorials, and if you've got the jar file in the right folder everything is runnable. For example, this is our intro classification tutorial - https://github.com/oracle/tribuo/blob/main/tutorials/irises-tribuo-v4.ipynb.

I am wondering what is the best / most cost-efficient way to serve llama V2. - llama.cpp (is it production ready or just for playing around?) ? - Triton inference server ? - HF text generation inference ?

For inference of CNNs, memory should really not be an issue. If it is a software engineering problem, not a hardware issue. FP16 or Int8 for weights is fine and weight size won’t increase due to the high resolution. And during inference memory used for hidden layer tensors can be reused as soon as the last consumer layer has been processed. You likely using something that is designed for training for inference and that blows up the memory requirement, or if you are using TensorRT or something like that, you need to be careful to avoid that every tasks loads their own copy of the library code into the GPU. Maybe look at https://github.com/triton-inference-server/server

I am looking for something like [Triton Inference Server](https://github.com/triton-inference-server/server) or [TFX Serving](https://www.tensorflow.org/tfx/guide/serving), but in Rust. I came across [Orkon](https://github.com/vertexclique/orkhon) which seems to be dormant and a bunch of examples off of the [Awesome-Rust-MachineLearning](https://github.com/vaaaaanquish/Awesome-Rust-MachineLearning)

You've already mentioned Seldon Core which is well worth looking at but if you're just after the raw multi-model serving aspect rather than a fully-fledged deployment framework you should maybe take a look at the individual inference servers: Triton Inference Server and MLServer both support multi-model serving for a wide variety of frameworks (and custom python models). MLServer might be a better option as it has an MLFlow runtime but only you will be able to decide that. There also might be other inference servers that do MMS that I'm not aware of.

[1] https://docs.nvidia.com/launchpad/ai/chatbot/latest/chatbot-triton-overview.html[2] https://github.com/triton-inference-server/server[3] https://neptune.ai/blog/deploying-ml-models-on-gpu-with-kyle-morris[4] https://thechief.io/c/editorial/comparison-cloud-gpu-providers/[5] https://geekflare.com/best-cloud-gpu-platforms/

"TensorFlow has the better deployment infrastructure"

Tensorflow Serving is nice in that it's so tightly integrated with Tensorflow. As usual that goes both ways. It's so tightly coupled to Tensorflow if the mlops side of the solution is using Tensorflow Serving you're going to get "trapped" in the Tensorflow ecosystem (essentially).

For pytorch models (and just about anything else) I've been really enjoying Nvidia Triton Server[0]. Of course it further entrenches Nvidia and CUDA in the space (although you can execute models CPU only) but for a deployment today and the foreseeable future you're almost certainly going to be using a CUDA stack anyway.

Triton Server is very impressive and I'm always surprised to see how relatively niche it is.

[0] - https://github.com/triton-inference-server/server