pytorch striking gradio using tensorflow with ArtEmis and JaX in Neo-x