Skip to content

[SR-12148] Make differentiation work with class types #52130

@rxwei

Description

@rxwei
Previous ID SR-12148
Radar rdar://problem/59278218
Original Reporter @rxwei
Type New Feature
Status In Progress
Resolution
Additional Detail from JIRA
Votes 0
Component/s
Labels New Feature
Assignee @dan-zheng
Priority Medium

md5: a71ffde72bc884ae0041737d20f410bc

Sub-Tasks:

blocks:

  • TF-641 [AD] Handle TangentVector classes in differentiation transform

relates to:

  • TF-38 KeyPathIterable derivation: Should work on classes
  • TF-481 Support mutable layers

Issue Description:

Differentiation should work with class types. This enables use cases like "mutable layers", which can be neatly defined as Differentiable-conforming classes with a func callAsFunction method.

Metadata

Metadata

Assignees

Labels

AutoDifffeatureA feature request or implementation

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions