Code health: torch.autograd Function, torch zeros, jax version - Githubissues

microsoft / knossos-ksc

Compiler with automatic differentiation

Other

45 stars 10 forks source link

Code health: torch.autograd Function, torch zeros, jax version #945

Closed awf closed 3 years ago

awf commented 3 years ago

Remove various warnings, add TODOs.

After:

Before (subset, I didn't capture all the JAX warnings)