There is no JAX container in DLFW release 24.05.
JAX Release 24.05
NVIDIA Optimized Frameworks (Latest Release)
Download PDF