Skip to main content
Back to top
Ctrl
+
K
kauldron
kauldron
Guides
Training
Eval and determinism
Checkpoint
Konfig - Demo
Klinen - Torch-like API for flax
Links
GitHub
API
kd
kd.ckpts
kd.ckpts.Checkpointer
kd.ckpts.NoopCheckpointer
kd.ckpts.AbstractPartialLoader
kd.ckpts.CkptSource
kd.ckpts.KauldronSource
kd.ckpts.PartialLoader
kd.ckpts.PyTreeCheckpointer
kd.data
kd.data.loaders
kd.data.loaders.GrainTfds
kd.data.loaders.ExtractRgbImages
kd.data.loaders.Tfds
kd.data.utils
kd.data.utils.array_spec_to_jnp_empty
kd.data.utils.drop_first_axis
kd.data.utils.mock_batch_from_elem_spec
kd.data.utils.Args
kd.data.utils.Kwargs
kd.data.utils.get_model_inputs
kd.data.TFDataPipeline
kd.data.CenterCrop
kd.data.Elements
kd.data.ElementWiseRandomTransform
kd.data.ElementWiseTransform
kd.data.HStack
kd.data.InceptionCrop
kd.data.OneHot
kd.data.PadFirstDimensionToFixedSize
kd.data.RandAugment
kd.data.RandomCrop
kd.data.RandomFlipLeftRight
kd.data.Rearrange
kd.data.Resize
kd.data.ResizeSmall
kd.data.TreeFlattenWithPath
kd.data.ValueRange
kd.data.VStack
kd.konfig
kd.konfig.ConfigDict
kd.konfig.register_aliases
kd.konfig.resolve
kd.konfig.imports
kd.konfig.mock_modules
kd.konfig.ref_copy
kd.konfig.ref_fn
kd.konfig.WithRef
kd.konfig.placeholder
kd.losses
kd.losses.compute_losses
kd.losses.Loss
kd.losses.L1
kd.losses.L2
kd.losses.SigmoidBinaryCrossEntropy
kd.losses.SoftmaxCrossEntropy
kd.losses.SoftmaxCrossEntropyWithIntLabels
kd.metrics
kd.metrics.Metric
kd.metrics.State
kd.metrics.TreeMap
kd.metrics.TreeReduce
kd.metrics.Accuracy
kd.metrics.RocAuc
kd.metrics.Ari
kd.metrics.Psnr
kd.metrics.Ssim
kd.metrics.LpipsVgg
kd.metrics.Norm
kd.metrics.Std
kd.nn
kd.nn.Module
kd.nn.train_property
kd.nn.interms_property
kd.nn.Dropout
kd.nn.Identity
kd.random
kd.random.PRNGKey
kd.summaries
kd.summaries.HistogramSummary
kd.summaries.ImageSummary
kd.summaries.PerImageChannelPCA
kd.summaries.ShowDifferenceImages
kd.summaries.ShowImages
kd.summaries.ShowSegmentations
kd.summaries.Summary
kd.train
kd.train.Config
kd.train.MultiEvaluator
kd.train.NoopEvaluator
kd.train.SingleEvaluator
kd.train.RngStream
kd.train.RngStreams
kd.train.train
kd.train.TrainState
kd.train.TrainStep
kd.typing
kd.typing.Annotated
kd.typing.Any
kd.typing.Callable
kd.typing.Hashable
kd.typing.Sequence
kd.typing.Union
kd.typing.ArraySpec
kd.typing.ElementSpec
kd.typing.PyTree
kd.typing.Array
kd.typing.Bool
kd.typing.Float
kd.typing.Float32
kd.typing.Int
kd.typing.Integer
kd.typing.Scalar
kd.typing.ScalarFloat
kd.typing.ScalarInt
kd.typing.TfArray
kd.typing.TfFloat
kd.typing.TfFloat32
kd.typing.TfInt
kd.typing.TfUInt8
kd.typing.UInt32
kd.typing.UInt8
kd.typing.Memo
kd.typing.Shape
kd.typing.TypeCheckError
kd.typing.typechecked
kd.typing.PRNGKey
kd.typing.PRNGKeyLike
kd.typing.Dtype
kd.typing.Key
kd.typing.Initializer
kd.typing.Axes
kd.typing.AxisName
kd.typing.Schedule
kd.inspect
kd.inspect.undo_process_input
kd.inspect.json_spec_like
kd.inspect.eval_context_shape
kd.inspect.get_colab_model_overview
kd.plotting
kd.plotting.plot_schedules
kd.xm
kd.xm.xmanager_api
kd.xm.RunConfig
kd.xm.Experiment
.md
.pdf
kauldron
kauldron
#
This is not an officially supported Google product.