Use `@differentiable(vjp:)` to work around a linker error.
Use `@differentiable(vjp:)` to work around a linker error.

tensorflow/swift-models,tensorflow/swift-models,tensorflow/swift-models,tensorflow/swift-models