[JAX FE] : Implement jax.lax.iota operation #16099

Job	Run time
Smart_CI	27s
Docker	53s
Build / Build	12m 13s
Samples	0s
C++ unit tests / C++ unit tests	59s
Debian Packages / Debian Packages	1m 38s
Python API tests / Python API tests	0s
Python unit tests / Python unit tests	3m 3s
JAX Layer Tests / JAX Layer Tests	2m 5s
OpenVINO tokenizers extension / OpenVINO tokenizers extension	2m 37s
Pytorch Layer Tests / PyTorch Layer Tests	2m 34s
TensorFlow Layer Tests / TensorFlow Layer Tests	13m 35s
ci/gha_overall_status_ubuntu_24	1s
	40m 5s

Provide feedback