As far as tech skills, there isn't a one size fits all answer -- the landscape, even within "early-stage startups in the AI space", is simply too broad.
You can get a good idea of what tools are most commonly used in the area just by browsing the web, looking at blog posts and open source projects, etc.
More specifically, I recommend going to LinkedIn or whatever job posting site you prefer and finding a bunch (10-20?) job postings that really excite you and tallying up the most commonly required skills. Go learn those, build up a portfolio of online demos and git repos, and you'll probably be as prepared as you can be to navigate the job market.
Source: started my career in early stage startups, now at Google. I even got lucky to have made a couple bucks from my equity (so I am a massive outlier, statistically speaking), but it still wasn’t worth it.
My biggest negative on the engineering side was that we were almost always encouraged to neglect good, testable architecture in favor of faster releases. This lead to more production breakages and confusion about what is affected by what changes.
Since then, I've realized just how critical testing is to a system, and that it's not a "nice to have", it's a must, especially as your team doubles in size again and again.
I had a technical founder who didn't lead the engineering team always say "we need more tests" (unit specifically), while not allowing us to refactoring our codebase to actually make it testable (most of the API code was just regular PHP scripts that could span 1k+ lines of business logic, untested).
That's another thing: your technical founder of head of the engineering isn't always going to be the smartest person ever. I had that idea in my head that of course they'd be an incredible engineer since they're the one in charge, but that isn't always the case. There's nothing wrong with that either, because hopefully they're a strong leader and that matters more for their position.
I do want to say though, I've worked at other earlier stage startups that have much more attention put into design decisions and testing and it is much more my speed. I can't fathom not having an automated test suite that tests the majority of my codebase automatically to catch the little stuff that bites you later on.
Whether or not that aligns with your own benefit is a different question entirely. If your comp is set up well and the company is supportive, it should be.