flax
Implement vjp and jvp transforms
#1629
Merged

Commits
  • Implement vjp and jvp transforms
    jheek committed 4 years ago
  • Fix pytype issue
    jheek committed 4 years ago
  • Fix docs
    jheek committed 4 years ago
Loading