Commits

Branch selector

 main

User selector

marcvanzee

Datepicker

All time

Commit History

Commits on Aug 26, 2022

• [jax2tf] Improves support for lax.gather (enable_xla=False).

  Show description for 6dd425c
  

  marcvanzee

  authored and

  jax authors

  committed
  6dd425c

  Copy full SHA for 6dd425c

Commits on Aug 22, 2022

• Update no_xla_limitations.md

  marcvanzee

  authored
  4f78a1d

  Copy full SHA for 4f78a1d

Commits on Aug 16, 2022

• [jax2tf] lax.reduce_window (enable_xla=False): bug fix and improvements.

  Show description for df5f3c5
  

  marcvanzee

  authored and

  jax authors

  committed
  df5f3c5

  Copy full SHA for df5f3c5

Commits on Aug 11, 2022

• [jax2tf] Adds more support for reduce_window when enable_xla=False.

  Show description for 15d3dc8
  

  marcvanzee

  authored and

  jax authors

  committed
  15d3dc8

  Copy full SHA for 15d3dc8

Commits on Aug 9, 2022

• [jax2tf] Fixes a bug in testing.

  Show description for 856d91d
  

  marcvanzee

  authored and

  jax authors

  committed
  856d91d

  Copy full SHA for 856d91d

Commits on Jul 14, 2022

• [jax2tf] Do not use tf.math.invert_permutation when enable_xla=False.

  Show description for 86ab8a8
  

  marcvanzee

  authored and

  jax authors

  committed
  86ab8a8

  Copy full SHA for 86ab8a8

Commits on Jul 12, 2022

• [jax2tf] Use tf.nest.flatten instead of tf.keras.backend.flatten.

  Show description for 9b54660
  

  marcvanzee

  authored and

  jax authors

  committed
  9b54660

  Copy full SHA for 9b54660

Commits on Jul 7, 2022

• Do not normalize FFT by a constant "1" if no normalization is provided (i.e., norm is None).

  Show description for 9d18f43
  

  marcvanzee

  authored and

  jax authors

  committed
  9d18f43

  Copy full SHA for 9d18f43

Commits on Jun 24, 2022

• Simplifies inverting permutation.

  Show description for 4c25ef1
  

  marcvanzee

  authored and

  jax authors

  committed
  4c25ef1

  Copy full SHA for 4c25ef1

Commits on Jun 3, 2022

• [jax2tf] Rewrites the converters_eval framework and makes various improvements.

  Show description for 1ecfe05
  

  marcvanzee

  authored and

  jax authors

  committed
  1ecfe05

  Copy full SHA for 1ecfe05

Commits on Apr 13, 2022

• [jax2tf] Fixes conv bug. The filter dims are allowed to be bigger than the input dims if the padding type is "SAME".

  Show description for af89426
  

  marcvanzee

  authored and

  jax authors

  committed
  af89426

  Copy full SHA for af89426

Commits on Jan 31, 2022

• Updates flax.nn with flax.linen in jax2tf README

  Show description for c52f5ea
  

  marcvanzee

  authored and

  jax authors

  committed
  c52f5ea

  Copy full SHA for c52f5ea

Commits on Dec 14, 2021

• Adds rtol to jax2tf/converters_eval tests.

  Show description for 0404dbd
  

  marcvanzee

  authored and

  jax authors

  committed
  0404dbd

  Copy full SHA for 0404dbd

Commits on Dec 9, 2021

• Finish jax2tf no_xla documentation

  marcvanzee

  committed
  9bd117f

  Copy full SHA for 9bd117f

Commits on Dec 8, 2021

• Restructures jax2tf converters eval directory.

  Show description for 3264045
  

  marcvanzee

  authored and

  jax authors

  committed
  3264045

  Copy full SHA for 3264045

Commits on Dec 6, 2021

• Adds support for batch_group_count in jax2tf. This was a request from a jax2tf user. I removed the jax2tf limitation which re-enables the test, which is passing now.

  Show description for 6f865b8
  

  marcvanzee

  authored and

  jax authors

  committed
  6f865b8

  Copy full SHA for 6f865b8

Commits on Nov 22, 2021

• Merge branch 'main' of http://www.github.com/google/jax into compare

  marcvanzee

  committed
  6eca387

  Copy full SHA for 6eca387

• Improve logics for numerical comparison

  marcvanzee

  committed
  7b25e05

  Copy full SHA for 7b25e05

Commits on Nov 18, 2021

• Merge upstream main

  marcvanzee

  committed
  fda30ad

  Copy full SHA for fda30ad

Commits on Nov 17, 2021

• Add numerical comparison with TF

  marcvanzee

  committed
  06d5e99

  Copy full SHA for 06d5e99

Commits on Nov 15, 2021

• Small fixes

  marcvanzee

  committed
  f6282c7

  Copy full SHA for f6282c7

• Change random data and updates table

  marcvanzee

  committed
  8bb2c76

  Copy full SHA for 8bb2c76

• Update PPO shapes fixing numerical diff

  marcvanzee

  committed
  775236a

  Copy full SHA for 775236a

• Fix incorrect param ordering

  marcvanzee

  committed
  2c17076

  Copy full SHA for 2c17076

• Merge branch 'main' of http://www.github.com/google/jax into compare

  marcvanzee

  committed
  f2d52f8

  Copy full SHA for f2d52f8

Commits on Nov 11, 2021

• Add numerical comparison for jax2tf -> TFLite

  marcvanzee

  committed
  3df11ae

  Copy full SHA for 3df11ae

Commits on Nov 2, 2021

• Adds support for batch dimensions for lax.gather when enable_xla=False. This allows us to, for instance, convert jax.vmap(lax.dynamic_slice).

  Show description for e28bb23
  

  marcvanzee

  authored and

  jax authors

  committed
  e28bb23

  Copy full SHA for e28bb23

Commits on Oct 22, 2021

• Implements a lowering function for tf.expm1 and adds tests.

  Show description for 0f47712
  

  marcvanzee

  authored and

  jax authors

  committed
  0f47712

  Copy full SHA for 0f47712

Commits on Oct 21, 2021

• Internal change

  Show description for 7a5e843
  

  marcvanzee

  authored and

  jax authors

  committed
  7a5e843

  Copy full SHA for 7a5e843

Commits on Oct 20, 2021

• Fixes a dtype bug in the conversion of dynamic_slice when enable_xla=False.

  Show description for 1b80fee
  

  marcvanzee

  authored and

  jax authors

  committed
  1b80fee

  Copy full SHA for 1b80fee

• Implements padding support for lax.reduce_window when enable_xla=False.

  Show description for 8dbd51d
  

  marcvanzee

  authored and

  jax authors

  committed
  8dbd51d

  Copy full SHA for 8dbd51d

Commits on Oct 18, 2021

• Factors out all enable_xla=False ops into a separate library impl_no_xla.py.

  Show description for 46b9653
  

  marcvanzee

  authored and

  jax authors

  committed
  46b9653

  Copy full SHA for 46b9653

Commits on Oct 15, 2021

• Improves support for conv_general_dilated in JAX for models running on the web or mobile through TFLite and TFjs.

  Show description for aaf3bb7
  

  marcvanzee

  authored and

  jax authors

  committed
  aaf3bb7

  Copy full SHA for aaf3bb7

Commits on Oct 11, 2021

• Implements a modular and easily extensible evaluation framework for both TFLite and TFjs. The evaluation framework has the following features:

  Show description for 161363d
  

  marcvanzee

  authored and

  jax authors

  committed
  161363d

  Copy full SHA for 161363d

Commits on Jun 22, 2021

• Implements dynamic_update_slice when enable_xla=False and adds tests.

  Show description for 1bba5d7
  

  marcvanzee

  authored and

  jax authors

  committed
  1bba5d7

  Copy full SHA for 1bba5d7

Pagination

Previous

Next