According to
this F-35B Model testing was completed in 2016. Indeed, it's a given that such a threat is considered in the design. Without proof, I'd assume the more numerous, and more exported F-35A would recieve no less attention.
That said, the deployment of nuclear weapons to a theatre changes the game in ways bigger than a single model of plane could have much bearing on.