Installing NVIDIA developer tools to base jax image

[terrykong]

Adding developer tools can help users with profiling and improving perf

### Tasks
- [ ] [Nsight](https://docs.nvidia.com/nsight-systems/InstallationGuide/index.html)

[nouiz]

nsys is now installed.