8000 Commits · gnecula/jax · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on Feb 2, 2021

Minor cleanup of limitations; regenerate docs
gnecula
committed

Commits on Jan 26, 2021

Add support for sharded_jit conversion to jax2tf.

gnecula
authored and
jax authors
committed

Commits on Jan 18, 2021

[jax2tf] Reflect in the limitations that add is now implemented for uint32 in TF
gnecula
committed
[jax2tf] Start using jit_compile instead of the deprecated experimental_compile
gnecula
committed
Copybara import of the project:

gnecula
authored and
jax authors
committed
Copybara import of the project:

gnecula
authored and
jax authors
committed

Commits on Jan 15, 2021

Commits on Jan 14, 2021

[jax2tf] Adjusting test tolerances

gnecula
authored and
jax authors
committed
[host_callback] Fix bug when id_tap takes no arguments but has a result.
gnecula
committed
[loops] Extend loops with support for pytrees
gnecula
committed

Commits on Jan 13, 2021

[host_callback] Refactor to merge the implementations of id_tap and call
gnecula
committed

Commits on Jan 7, 2021

Commits on Jan 6, 2021

Commits on Jan 5, 2021

Merge remote-tracking branch 'origin/print_refactor' into print_refactor
gnecula
committed

0