search

SmilingWolf / JAX-CV

Repo for my JAX CV experiments. Mostly revolving around the Danbooru20xx dataset
24 stars 4 forks source link

Remove usage of power(x, -0.5) #2

Closed SmilingWolf closed 1 year ago

SmilingWolf commented 1 year ago

JAX is smart enough to rewrite it as rsqrt where necessary on its own