AI frameworks for Jetson Xavier NX

Can any list all of the AI Frameworks that the Xavier NX can be used to train models on. I am only able to find Tensorflow, PyTorch, MXNet, Caffe/Caffe2, and Keras.

Hi @rltrozzi, in addition to those I have seen people install DarkNet and ONNXRuntime (although ONNXRuntime is for inferencing). Since JetPack includes cuDNN, it may be possible to compile/install other frameworks as well on Jetson - those you listed just seem to be the most popular. Is there something in particular you are looking for?

would it be able to run/support h2o, Auto-sklearn, tpot, and autogluon?

I’m not personally familiar with those, but during a quick check of h2o, I was simply able to pip3 install h2o inside of the l4t-ml container without issues. auto-sklearn appears to be installing as well following their install instructions. You might need to include this suggestion when installing the auto-sklearn prerequisites.