Package-level declarations

Types

Link copied to clipboard
object Losses

Java-friendly factory for loss functions.

Link copied to clipboard
object Optimizers

Java-friendly factory for optimizers.

Link copied to clipboard
class SequentialModelBuilder @JvmOverloads constructor(ctx: ExecutionContext, dtype: DType = FP32)

Java-friendly sequential model builder that mirrors the Kotlin DSL.

Link copied to clipboard

Java-friendly static utility class wrapping the most common tensor operations.

Link copied to clipboard

Java-friendly training loop wrapping SKaiNET's trainStep function.

Link copied to clipboard
data class TrainingResult(val epochs: Int, val finalLoss: Float)

Result from a training run.