search

guanyilun / galilei

A generic function emulator that supports multiple backends
MIT License
1 stars 0 forks source link

jax: improve loss calculation #16

Closed guanyilun closed 1 year ago

guanyilun commented 1 year ago

improve jax loss function calculation.