Patrick von Platen fbf468b057 [Flax] Correct flax docs (#12782) ...

* fix_torch_device_generate_test

* remove @

* fix flax docs

* correct more docs in flax

* another correction

* fix flax docs

* Apply suggestions from code review

2021-08-04 16:31:23 +02:00

9.5 KiB

Raw Blame History

View Raw