unilab.algos.mlx.common.activations

Activation helpers for MLX models.

Functions

get_activation(name)

Resolve a string activation name to a callable.
unilab.algos.mlx.common.activations.get_activation(name)[source]

Resolve a string activation name to a callable.

Parameters:

name (str | None)

Return type:

Callable[[array], array]