Join GitHub today
GitHub is home to over 36 million developers working together to host and review code, manage projects, and build software together.Sign up
[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: