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

Job	Run time
Smart_CI	20s
Docker	46s
Build / Build	5m 46s
Debian Packages	0s
ONNX Runtime Integration	0s
Samples	0s
OpenVINO JS API	0s
PyTorch Models tests	0s
CPU functional tests	0s
C++ unit tests / C++ unit tests	41s
JAX Layer Tests / JAX Layer Tests	1m 18s
OpenVINO tokenizers extension / OpenVINO tokenizers extension	1m 45s
Python API tests / Python API tests	0s
Python unit tests / Python unit tests	2m 14s
Pytorch Layer Tests / PyTorch Layer Tests	2m 25s
TensorFlow Models tests	0s
TensorFlow Layer Tests / TensorFlow Layer Tests	3m 45s
ci/gha_overall_status_linux_arm64	1s
	19m 1s

Provide feedback