JAX images update #967
Description
Description
It seems like current JAX images and support of it here is a little outdated.
Any plans on the matter?
Describe the solution you'd like
Ready to go JAX docker image and updated documentation/
Describe alternatives you've considered
Taking another 1API docker image as a base one and completing the environment manually for JAX. (But here's no related documentation as well).