NASimJax: A GPU-Accelerated Policy Learning Framework for Penetration Testing
NASimJax is a JAX-based framework that accelerates penetration testing policy learning by up to 100x throughput compared to original simulators. It reformulates testing as a Contextual POMDP and uses novel network generation pipelines to create diverse, solvable scenarios. This allows for principled study of zero-shot policy generalization across unseen networks.