Commits

Branch selector

 a5f906a

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 14, 2020

• jax2tf import fix

  mattjj

  committed
  a5f906a

  Copy full SHA for a5f906a

• add iota_p support to jax2tf

  mattjj

  committed
  b99e350

  Copy full SHA for b99e350

Commits on Oct 11, 2020

• fix type error in jax2tf

  mattjj

  committed
  b402b87

  Copy full SHA for b402b87

• don't generate lazy iota/eye/tri/delta omnistaging

  mattjj

  committed
  4e65a6f

  Copy full SHA for 4e65a6f

Commits on Oct 10, 2020

• Merge pull request jax-ml#4532 from gnecula:tf_omnistaging

  Show description for a97cb87
  

  jax authors

  committed
  a97cb87

  Copy full SHA for a97cb87

• [jax2tf] Ensure jax2tf still works without omnistaging

  Show description for 9d9762b
  gnecula

  committed
  9d9762b

  Copy full SHA for 9d9762b

Commits on Oct 9, 2020

• Merge pull request jax-ml#4065 from skye:wheels

  Show description for 0d69a84
  

  jax authors

  committed
  0d69a84

  Copy full SHA for 0d69a84

• Use local version identifiers to distribute cuda jaxlib wheels.

  Show description for cacb017
  skye

  committed
  cacb017

  Copy full SHA for cacb017

• Merge pull request jax-ml#4524 from jakevdp:fix-mean

  Show description for ae3d310
  

  jax authors

  committed
  ae3d310

  Copy full SHA for ae3d310

• Merge pull request jax-ml#4526 from google:4510-2

  Show description for cd03bea
  

  jax authors

  committed
  cd03bea

  Copy full SHA for cd03bea

• fix x64 issue

  mattjj

  committed
  78fe038

  Copy full SHA for 78fe038

• add test for scan input forwarding

  mattjj

  committed
  0aeeb63

  Copy full SHA for 0aeeb63

• make scan fwd raw extensive inputs as DeviceArray

  Show description for 5355776
  mattjj

  committed
  5355776

  Copy full SHA for 5355776

• Merge pull request jax-ml#4470 from gnecula:tf_tracers

  Show description for fc69d45
  

  jax authors

  committed
  fc69d45

  Copy full SHA for fc69d45

• Fix extraneous dtype warning in jnp.mean

  jakevdp

  committed
  67a77c7

  Copy full SHA for 67a77c7

• [jax2tf] Port jax2tf to use omnistaging

  Show description for 0213efd
  gnecula

  committed
  0213efd

  Copy full SHA for 0213efd

• Merge pull request jax-ml#4518 from gnecula:tf_debug

  Show description for e194dff
  

  jax authors

  committed
  e194dff

  Copy full SHA for e194dff

• [jax2tf] Fix the behavior for shift when the shift amount is negative or large.

  Show description for a1e67be
  gnecula

  committed
  a1e67be

  Copy full SHA for a1e67be

• Merge pull request jax-ml#4517 from google:issue4510

  Show description for e6399cd
  

  jax authors

  committed
  e6399cd

  Copy full SHA for e6399cd

• make pytype happy

  mattjj

  committed
  7691be9

  Copy full SHA for 7691be9

• optimize scan partial_eval to fix jax-ml#4510

  Show description for 52fe026
  mattjj

  committed
  52fe026

  Copy full SHA for 52fe026

Commits on Oct 8, 2020

• Merge pull request jax-ml#4514 from google:remove-unneeded-code

  Show description for d4da9cc
  

  jax authors

  committed
  d4da9cc

  Copy full SHA for d4da9cc

• Merge pull request jax-ml#4515 from google:enable-conv-test

  Show description for ab462ba
  

  jax authors

  committed
  ab462ba

  Copy full SHA for ab462ba

• Merge pull request jax-ml#4512 from jakevdp:raise-to-shaped

  Show description for 4994b82
  

  jax authors

  committed
  4994b82

  Copy full SHA for 4994b82

• re-enable lax autodiff test

  Show description for 1614e28
  froystig

  committed
  1614e28

  Copy full SHA for 1614e28

• remove mysterious code that is no longer needed

  mattjj

  committed
  55be9dd

  Copy full SHA for 55be9dd

• Merge pull request jax-ml#4511 from google:debug-nans-test-move

  Show description for 91492ca
  

  jax authors

  committed
  91492ca

  Copy full SHA for 91492ca

• lax_control_flow: retry via function rather than via loop

  jakevdp

  committed
  ef25da8

  Copy full SHA for ef25da8

• Fallback to Python for float0 in the C++ jax.jit.

  Show description for 008eae6
  

  jblespiau

  authored and

  jax authors

  committed
  008eae6

  Copy full SHA for 008eae6

• move a debug_nans test into debug_nans test file

  mattjj

  committed
  24de811

  Copy full SHA for 24de811

• Merge pull request jax-ml#4161 from jakevdp:raise-to-shaped

  Show description for ddcfd45
  

  jax authors

  committed
  ddcfd45

  Copy full SHA for ddcfd45

• Merge pull request jax-ml#4483 from google:debug-nans-no-store-error

  Show description for 3be8188
  

  jax authors

  committed
  3be8188

  Copy full SHA for 3be8188

• Merge pull request jax-ml#4483 from google:debug-nans-no-store-error

  Show description for 57ec9dd
  

  jax authors

  committed
  57ec9dd

  Copy full SHA for 57ec9dd

• wait for result in debug_nans_test

  mattjj

  committed
  09f2be1

  Copy full SHA for 09f2be1

• Merge pull request jax-ml#4508 from zhangqiaorjc:expm_test_slow

  Show description for 2b76bcc
  

  jax authors

  committed
  2b76bcc

  Copy full SHA for 2b76bcc

Pagination

Previous

Next