install pip install brax-jumpy and then import jumpy as jp don't import jumpy' from 'brax' again otherwise you will receive the same error.