unilab.algos.mlx.common.activations.get_activation

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]