mirror of
https://github.com/velocitatem/PHANTOM.git
synced 2026-05-31 16:43:36 +00:00
14 lines
281 B
Python
14 lines
281 B
Python
"""JAX-compatible training and environment modules for PHANTOM."""
|
|
|
|
from __future__ import annotations
|
|
|
|
try:
|
|
import jax # noqa: F401
|
|
import jax.numpy as jnp # noqa: F401
|
|
|
|
JAX_AVAILABLE = True
|
|
except ImportError:
|
|
JAX_AVAILABLE = False
|
|
|
|
__all__ = ["JAX_AVAILABLE"]
|