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