Join GitHub today
[WIP] SameDiff losses/differentiation overhaul #7452
Allows users to specify which variables represent the losses. Also differentiates only the subset of the graph that is actually needed (more efficient). Consequently, it's now possible to train only the subset of the graph that the loss function depends on.
Some other major changes: