PyTorch
API: torch.lbeta
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'
Keras
API: —
Strategy: Macro 'keras.ops.sum(keras.ops.log_gamma({x}), axis=-1) - keras.ops.log_gamma(keras.ops.sum({x}, axis=-1))'
TensorFlow
API: tf.math.lbeta
Strategy: Direct Mapping
Flax NNX
API: —
Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'
PaxML / Praxis
API: —
Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'