Package-level declarations

Types

Link copied to clipboard
@Target(allowedTargets = [AnnotationTarget.FUNCTION])
annotation class Diff(val ruleName: String = "")

Annotation to mark TensorOps methods that require an adjoint (gradient) implementation.

Link copied to clipboard
@Target(allowedTargets = [AnnotationTarget.CLASS])
annotation class GenerateTracingWrapper

Annotation to mark interfaces for automatic tracing wrapper generation via KSP.