Commits

Branch selector

 main

User selector

maxwillzq

Datepicker

All time

Commit History

Commits on Apr 5, 2023

• Update unit test and doc how work around jit_compile=False on TPU for native_serialization.

  Show description for 0e549ac
  

  maxwillzq

  authored and

  jax authors

  committed
  0e549ac

Commits on Mar 24, 2023

• [1/n] store embedded tf.graph to stablehlo.custom_call

  Show description for fad4e6f
  

  maxwillzq

  authored and

  jax authors

  committed
  fad4e6f

Commits on Mar 23, 2023

• Add padding option "SAME_LOWER" for ticket jax-ml#14990

  Show description for 171b22d
  

  maxwillzq

  authored and

  jax authors

  committed
  171b22d

Commits on Feb 17, 2023

• Add tf.convert_to_tensor for call_tf gradient outputs.

  Show description for fdc8864
  

  maxwillzq

  authored and

  jax authors

  committed
  fdc8864

Commits on Feb 16, 2023

• Fix the simple bug on call_tf.replace_non_float and add unittest for floating and complex data type.

  Show description for d0b42f2
  

  maxwillzq

  authored and

  jax authors

  committed
  d0b42f2

Commits on Feb 14, 2023

• clean up no-need exception raise.

  Show description for 4237f22
  

  maxwillzq

  authored and

  jax authors

  committed
  4237f22

Commits on Feb 10, 2023

• Enable call_tf_native_lowering_test.

  Show description for 7659a3a
  

  maxwillzq

  authored and

  jax authors

  committed
  7659a3a

Commits on Feb 1, 2023

• Consolidate the experimental_get_compiler_ir eager and tf function path in jax2tf.call_tf.

  Show description for 0cd3dee
  

  maxwillzq

  authored and

  jax authors

  committed
  0cd3dee

Commits on Jan 17, 2023

• Change args_tf_flat to tf.TensorSpec in jax2tf.call_tf.

  Show description for 469a8eb
  

  maxwillzq

  authored and

  jax authors

  committed
  469a8eb

Commits on Dec 6, 2022

• Cache the exec time tf.get_current_name_scope into _thread_local_state.exec_time_tf_name_scope and add it as prefix during tracing.

  Show description for 2fc7bbc
  

  maxwillzq

  authored and

  jax authors

  committed
  2fc7bbc

Commits on Nov 14, 2022

• Add tf.get_current_name_scope() as prefix of name_stack during tf execution.

  Show description for 7c6a65b
  

  maxwillzq

  authored and

  jax authors

  committed
  7c6a65b

Commits on Nov 9, 2022

• Internal change

  Show description for b41c594
  

  maxwillzq

  authored and

  jax authors

  committed
  b41c594

Commits on Oct 20, 2022

• fix jax2tf readme typo

  Show description for 408953b
  

  maxwillzq

  authored and

  jax authors

  committed
  408953b

Commits on Aug 31, 2022

• [jax2tf] add a new test for jax2tf gda test.

  Show description for 59c2fc9
  

  maxwillzq

  authored and

  jax authors

  committed
  59c2fc9

Commits on Aug 25, 2022

• Fix jax2tf GDA bug.

  Show description for 28b14b1
  

  maxwillzq

  authored and

  jax authors

  committed
  28b14b1

Commits on Jul 12, 2022

• Fix limitations-of-call_tf github link typo.

  Show description for 153b6ae
  

  maxwillzq

  authored and

  jax authors

  committed
  153b6ae

Commits on Jun 27, 2022

• update jax2tf README: add walkaround about tf.Module magic conversion.

  Show description for de464fc
  

  maxwillzq

  authored and

  jax authors

  committed
  de464fc