Object

Base class for NNX objects that are not Pytrees.

PyTorch

API: object
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.object
Strategy: Direct Mapping
Official Docs ↗

Keras

API: object
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: object
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: object
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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

PaxML / Praxis

API: object
Strategy: Direct Mapping
Official Docs ↗