"hello world! " program in python reinforcement learning environment

Job ID: 36769804

Budget: ₹4,000 – ₹6,000 INR

# write RL(Reinforcement learning) environment with just 3 steps and terminates on 4th step.
# Note: Code should be in jax(python jit)
# print "hello:1" in first step
# print "hello world:2" in second step
# print "hello world!:3" in third step
# thats all !
# use/call the functions defined in attached file to do this and no hard coding
# complete details in attachment
#experience in Jax is great but not needed, experience in AI/ML is needed.