Craftax: (Crafter and NetHack) RL Environment in Jax