kups.core.utils ¶

Core utilities for JAX-based scientific computing.

This package provides essential utilities for numerical computing with JAX, organized into the following modules:

• ema: Exponential moving average with numerical stability
• functools: Functional programming combinators (compose, curry, etc.)
• jax: JAX-specific utilities (jit, vectorize, dataclass, etc.)
• math: Mathematical functions (cubic roots, matrix operations)
• ops: Array operations with broadcasting utilities
• position: Particle position utilities for periodic systems
• quaternion: 3D rotation representation and operations

Most commonly used items can be imported directly from submodules.