# kd.knn

[[[Source]]](https://github.com/google-research/kauldron/tree/main/kauldron/klinen/__init__.py)

```{eval-rst}
.. automodule:: kauldron.klinen
  :no-members:
```

## Symbols


### Class

|  |  |
--- | ---
[kd.knn.Dense](Dense) | Dense(features: int, use_bias: bool = True, dtype: Union[str, type[Any], numpy.dtype, jax._src.typing.SupportsDType, Any, NoneType] = None, param_dtype: Union[str, type[Any], numpy.dtype, jax._src.typing.SupportsDType, Any] = <class 'jax.numpy.float32'>, precision: Union[NoneType, str, jax._src.lax.lax.Precision, tuple[str, str], tuple[jax._src.lax.lax.Precision, jax._src.lax.lax.Precision]] = None, kernel_init: Union[jax.nn.initializers.Initializer, collections.abc.Callable[..., Any]] = <function variance_scaling.<locals>.init at 0x76412019ad40>, bias_init: Union[jax.nn.initializers.Initializer, collections.abc.Callable[..., Any]] = <function zeros at 0x76410a396fc0>, promote_dtype: flax.linen.linear.PromoteDtypeFn = <function promote_dtype at 0x76410689fce0>, dot_general: collections.abc.Callable[..., typing.Union[jax.Array, typing.Any]] | None = None, dot_general_cls: Any = None, parent: Union[flax.linen.module.Module, flax.core.scope.Scope, flax.linen.module._Sentinel, NoneType] = <flax.linen.module._Sentinel object at 0x76410688a690>, name: Optional[str] = None, *, _kd_state: 'Optional[_ModuleState]' = None)
[kd.knn.Dropout](Dropout) | Dropout(rate: float, broadcast_dims: collections.abc.Sequence[int] = (), deterministic: bool | None = None, rng_collection: str = 'dropout', parent: Union[flax.linen.module.Module, flax.core.scope.Scope, flax.linen.module._Sentinel, NoneType] = <flax.linen.module._Sentinel object at 0x76410688a690>, name: Optional[str] = None, *, _kd_state: 'Optional[_ModuleState]' = None)
[kd.knn.Module](Module) | Base Module class.
[kd.knn.Sequential](Sequential) | Sequential(layers: collections.abc.Sequence[collections.abc.Callable[..., typing.Any]], parent: Union[flax.linen.module.Module, flax.core.scope.Scope, flax.linen.module._Sentinel, NoneType] = <flax.linen.module._Sentinel object at 0x76410688a690>, name: Optional[str] = None, *, _kd_state: 'Optional[_ModuleState]' = None)

### Function

|  |  |
--- | ---
[kd.knn.convert](convert) | Decorator that convert a flax class into klinen.

### Typing

|  |  |
--- | ---
[kd.knn.Intermediate](Intermediate) | 

```{toctree}
:hidden:

Dense
Dropout
Intermediate
Module
Sequential
convert
```