First time to Jax and need some recommendations please 🙏