If you're a tech employee in a large company with lucrative compensation, you should be aggressively reducing your expenses and banking your excess so you can weather what might be long period of unemployment and can adapt more smoothly to employment at more modest compensation when you manage to get back in.
Unless you're working very obviously outside the blast radius of an AI-bubble correction (you'd know if you were) or are a very high-value VIP (again, you'd know), you should assume you'll be spending some time without a job within the next few years. Possibly a long time.
You might get lucky, but it's not really going to be in your control and "milking the AI wave, learning all the tools" isn't going to change your odds much. It really is musical chairs. Whether you lose your job will depend on where you happen to be standing when the music stops. And there are going to be so many other people looking for the same new chair as you, with resumes that look almost exactly like yours, that getting a new job will basically come down to a lottery draw.
If you think the AI stuff is cool, study it and play with it. Otherwise, just save money and start working on the outline for that novel you've been thinking about writing.