Skip to main content

Ctrl+K

kauldron

kauldron

Guides

Introduction
Train, eval, randomness
Sharding & Model parallelism
Checkpoint
Data pipelines
Konfig principles

Modules

Konfig - Demo
kontext
PyGrain data pipeline
Metrics
Klinen - Torch-like API for flax
OO Random API
Type Annotations and Shape checking

Links

GitHub
Issues

API

kd

.md

kd.typing.ElementSpec

kd.typing.ElementSpec#

from clu.data.dataset_iterator import ArraySpec, ElementSpec, PyTree

previous

kd.typing.Dim

next

kd.typing.Float

By kauldron authors

© Copyright 2023, kauldron authors.