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