setup.py install not working

[stalkerg]

Basically, because you have git+ in your req (looks like it's not valid for setup_tools and only for pip)

error in vkJAX setup command: 'install_requires' must be a string or list of strings containing valid project/version requirement specifiers; Parse error at "'+git://g'": Expected stringEnd

[alexander-g]

Setup.py should be fixed now.

Thanks for taking interest in this repo, but I will not continue with it. It was a project mostly for fun and educational purposes. If you want GPU acceleration on different hardware I recommend converting your JAX functions to ONNX and use a specialized ONNX runtime.

[stalkerg]

Do you know ONNX runtime for kompute or just Vulkan?