chor: implementing prallelization across jax

This commit is contained in:
2026-03-10 17:05:16 +01:00
parent 6d9613c0b6
commit 974498dab2
5 changed files with 303 additions and 41 deletions

3
engine/jax/__init__.py Normal file
View File

@@ -0,0 +1,3 @@
from .robust import select_adversarial_alpha_jax, _JAX_OK
__all__ = ["select_adversarial_alpha_jax", "_JAX_OK"]