skainet-lang-ksp-annotations/sk.ainet.lang.trace/Diff

Diff

@Target(allowedTargets = [AnnotationTarget.FUNCTION])

annotation class Diff(val ruleName: String = "")(source)

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

When applied to a method in an interface annotated with @GenerateTracingWrapper, the KSP processor will ensure that this operation is tracked as differentiable.

Properties

ruleName

val ruleName: String