velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 08:33:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: init code for jax core

Browse Source
This commit is contained in:
Daniel Rosel
2026-01-22 13:10:15 +01:00
parent a217d53556
commit 40e0b201e6
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sim/rl/jax_core/__init__.py Normal file
View File

@@ -0,0 +1,11 @@
"""JAX-accelerated simulation core for PHANTOM environment."""
from .transitions import TransitionData, compile_transitions, fallback_transitions, JAX_AVAILABLE
from .simulation import SessionBatch, SimResult, sample_sessions, compute_metrics
from .features import session_features, compute_session_transitions
from .separability import compute_divergences, estimate_alpha_batch
__all__ = [
"JAX_AVAILABLE", "TransitionData", "compile_transitions", "fallback_transitions",
"SessionBatch", "SimResult", "sample_sessions", "compute_metrics",
"session_features", "compute_session_transitions", "compute_divergences", "estimate_alpha_batch",
]