PureState

Helper for tuple/state construction.

Abstract Signature:

PureState(args)

PyTorch

API: tuple
Strategy: Direct Mapping
Official Docs ↗

Keras

API: tuple
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tuple
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: tuple
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.PureState
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: tuple
Strategy: Direct Mapping
Official Docs ↗