Commits

Branch selector

 a0f6354

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 23, 2023

• Merge pull request jax-ml#14638 from hawkinsp:conda

  Show description for a0f6354
  

  jax authors

  committed
  a0f6354

  Copy full SHA for a0f6354

• Update conda GPU install instructions.

  hawkinsp

  committed
  eb1710f

  Copy full SHA for eb1710f

• [jax2tf] Enable the strict platform checks for native serialization

  Show description for c4778c4
  

  gnecula

  authored and

  jax authors

  committed
  c4778c4

  Copy full SHA for c4778c4

• [jax2tf] Use CUDA and ROCM instead of GPU for XlaCallModuleOp platforms

  Show description for c6956f7
  

  gnecula

  authored and

  jax authors

  committed
  c6956f7

  Copy full SHA for c6956f7

• Copybara import of the project:

  Show description for db97728
  

  gnecula

  authored and

  jax authors

  committed
  db97728

  Copy full SHA for db97728

• Use the standard jtu.create_global_mesh instead of creating a mesh from scratch.

  Show description for 0d834c0
  

  yashk2810

  authored and

  jax authors

  committed
  0d834c0

  Copy full SHA for 0d834c0

Commits on Feb 22, 2023

• Merge pull request jax-ml#14610 from sharadmv:state-effect

  Show description for 5b2d3d9
  

  jax authors

  committed
  5b2d3d9

  Copy full SHA for 5b2d3d9

• Export the Shard type.

  Show description for 893d359
  

  saeta

  authored and

  jax authors

  committed
  893d359

  Copy full SHA for 893d359

• Merge pull request jax-ml#14629 from hawkinsp:deviceput

  Show description for 661c9e1
  

  jax authors

  committed
  661c9e1

  Copy full SHA for 661c9e1

• Merge pull request jax-ml#14628 from hawkinsp:init

  Show description for dec9557
  

  jax authors

  committed
  dec9557

  Copy full SHA for dec9557

• Export device_put_p as jax.lax.device_put_p.

  Show description for 8a2765a
  hawkinsp

  committed
  8a2765a

  Copy full SHA for 8a2765a

• Slightly increase the tolerance in sparse tests to avoid flakiness

  Show description for 1638313
  

  apaszke

  authored and

  jax authors

  committed
  1638313

  Copy full SHA for 1638313

• Fix __module__ of jax.nn.initializers.* to be jax.nn.initializers.

  Show description for fd24f97
  hawkinsp

  committed
  fd24f97

  Copy full SHA for fd24f97

• Merge pull request jax-ml#14594 from gnecula:tf_platform_check

  Show description for 7e001d8
  

  jax authors

  committed
  7e001d8

  Copy full SHA for 7e001d8

• Hide accidental exports from jax.core.

  Show description for 1d4b7a3
  

  hawkinsp

  authored and

  jax authors

  committed
  1d4b7a3

  Copy full SHA for 1d4b7a3

• Update links from iree-org/iree to openxla/iree

  Show description for 32d11fd
  

  GMNGeoffrey

  authored and

  jax authors

  committed
  32d11fd

  Copy full SHA for 32d11fd

• Only check for _device if device_buffer is xc.Buffer.

  Show description for c880318
  

  yashk2810

  authored and

  jax authors

  committed
  c880318

  Copy full SHA for c880318

• Add JaxprInputEffect and refactor StateEffects to use it

  sharadmv

  committed
  a6c4c87

  Copy full SHA for a6c4c87

• Merge pull request jax-ml#14613 from google:libtpu_install

  Show description for 767388c
  

  jax authors

  committed
  767388c

  Copy full SHA for 767388c

Commits on Feb 21, 2023

• Add --pre to nightly libtpu pip install command.

  Show description for 1476a85
  skye

  committed
  1476a85

  Copy full SHA for 1476a85

• Delete PyShardedBuffer since it was only used for GDAs and GDA is deprecated.

  Show description for 8fd70cc
  

  yashk2810

  authored and

  jax authors

  committed
  8fd70cc

  Copy full SHA for 8fd70cc

• Return jax.Array with a single device sharding from make_device_array.

  Show description for be31356
  

  yashk2810

  authored and

  jax authors

  committed
  be31356

  Copy full SHA for be31356

• Remove global_device_array from shared jax bazel library.

  Show description for 0bb75af
  

  hawkinsp

  authored and

  jax authors

  committed
  0bb75af

  Copy full SHA for 0bb75af

• Merge pull request jax-ml#14593 from eltociear:patch-5

  Show description for 6414539
  

  jax authors

  committed
  6414539

  Copy full SHA for 6414539

• Merge pull request jax-ml#14567 from jakevdp:typing-doc

  Show description for ea6e820
  

  jax authors

  committed
  ea6e820

  Copy full SHA for ea6e820

• [jax2tf] Enable strict platform checking for native serialized modules.

  Show description for b8f96f0
  gnecula

  committed
  b8f96f0

  Copy full SHA for b8f96f0

• DOC: improve usage recommendation in jax.typing

  jakevdp

  committed
  de673ce

  Copy full SHA for de673ce

• Fix typo in maps.py

  Show description for 28f89f6
  eltociear

  authored
  28f89f6

  Copy full SHA for 28f89f6

• [jax2tf] Customize limitations for native lowering mode

  Show description for 9e209a1
  

  gnecula

  authored and

  jax authors

  committed
  9e209a1

  Copy full SHA for 9e209a1

Commits on Feb 19, 2023

• Remove *_mhlo compatibility shims from jaxlib

  Show description for f337c00
  

  Eugene Burmako

  authored and

  jax authors

  committed
  f337c00

  Copy full SHA for f337c00

Commits on Feb 18, 2023

• Rename in_axis_resources and out_axis_resources with in_shardings and out_shardings. This is just a simple name replacement. It does not change any of the current pjit semantics and doesn't…

  Show description for 418c2f9
  

  yashk2810

  authored and

  jax authors

  committed
  418c2f9

  Copy full SHA for 418c2f9

• Merge pull request jax-ml#14549 from sharadmv:dbidx-effects

  Show description for c0107cc
  

  jax authors

  committed
  c0107cc

  Copy full SHA for c0107cc

• Replace op_sharding_sharding with gspmd_sharding. This is purely an internal change.

  Show description for d93aa70
  

  yashk2810

  authored and

  jax authors

  committed
  d93aa70

  Copy full SHA for d93aa70

• Refactor effects system to use effect types, not objects

  sharadmv

  committed
  af2306c

  Copy full SHA for af2306c

• Rename jax.sharding.OpShardingSharding to jax.sharding.GSPMDSharding. jax.sharding.OpShardingSharding will be removed in 3 months from Feb 17, 2023.

  Show description for 0ffdeb3
  

  yashk2810

  authored and

  jax authors

  committed
  0ffdeb3

  Copy full SHA for 0ffdeb3

Pagination

Previous

Next