Commits

Branch selector

 master

User selector

skye

Datepicker

All time

Commit History

Commits on Apr 24, 2020

• Remove platform canonicalization from xla_bridge.py (jax-ml#2815)

  skye

  authored
  343e486

  Copy full SHA for 343e486

Commits on Apr 23, 2020

• Pin mypy version in .travis.yml. (jax-ml#2811)

  Show description for 9b6976b
  skye

  authored
  9b6976b

  Copy full SHA for 9b6976b

• Make ShardedDeviceArray._value threadsafe again. (jax-ml#2810)

  Show description for 4630649
  skye

  authored
  4630649

  Copy full SHA for 4630649

Commits on Apr 21, 2020

• Bump jaxlib version in README to 0.1.45

  skye

  committed
  6db1f0c

  Copy full SHA for 6db1f0c

• Bump jaxlib version to 0.1.45 and update WORKSPACE and CHANGELOG. (jax-ml#2785)

  skye

  authored
  4703b0f

  Copy full SHA for 4703b0f

Commits on Apr 17, 2020

• Update automatic jaxlib install command to use nvidia-smi instead of nvcc. (jax-ml#2758)

  Show description for a1b4fe4
  skye

  authored
  a1b4fe4

  Copy full SHA for a1b4fe4

Commits on Apr 16, 2020

• Fix some bugs in _reshape_sharded_device_array (jax-ml#2732)

  skye

  authored
  0e29bd4

  Copy full SHA for 0e29bd4

• Temporarily make ShardedDeviceArray.__init__ optionally accept old si… (jax-ml#2730)

  Show description for fe8d1d1
  skye

  authored
  fe8d1d1

  Copy full SHA for fe8d1d1

Commits on Apr 15, 2020

• Add type hint to fix pytype error. (jax-ml#2727)

  Show description for 7a61aea
  skye

  authored
  7a61aea

  Copy full SHA for 7a61aea

• Allow ShardedDeviceArrays to represent arbitrary data shardings. (jax-ml#2142)

  Show description for 07571ae
  skye

  authored
  07571ae

  Copy full SHA for 07571ae

Commits on Apr 14, 2020

• Instantiate RNG in testcase instead of test harness in a few more places. (jax-ml#2706)

  skye

  authored
  be6c5cb

  Copy full SHA for be6c5cb

• Fix packbits/unpackbits tests (jax-ml#2702)

  skye

  authored
  4e9f640

  Copy full SHA for 4e9f640

Commits on Apr 13, 2020

• Add --export_dir and --baseline_dir flags to benchmark.py. (jax-ml#2677)

  Show description for 8c2901c
  skye

  authored
  8c2901c

  Copy full SHA for 8c2901c

Commits on Apr 8, 2020

• Fix up previous jaxpr.rst commit. (jax-ml#2647)

  skye

  authored
  f37f235

  Copy full SHA for f37f235

• Update scan jaxpr documentation. (jax-ml#2641)

  Show description for f8dc650
  skye

  authored
  f8dc650

  Copy full SHA for f8dc650

• Fully stage out scan under jit. (jax-ml#2638)

  Show description for f0ab673
  skye

  authored
  f0ab673

  Copy full SHA for f0ab673

Commits on Apr 6, 2020

• Adjust pmap_bechmark.py values to be more realistic. (jax-ml#2622)

  skye

  authored
  3fe8bd0

  Copy full SHA for 3fe8bd0

Commits on Apr 2, 2020

• Revert jax.numpy.matmul to pre-jax-ml#2512 version. (jax-ml#2584)

  Show description for 5c0ac40
  skye

  authored
  5c0ac40

  Copy full SHA for 5c0ac40

Commits on Apr 1, 2020

• Revert "Refactor shard_args ShardedDeviceArray slow path. (jax-ml#2562)" (jax-ml#2577)

  Show description for 1bb9aaa
  skye

  authored
  1bb9aaa

  Copy full SHA for 1bb9aaa

• Refactor shard_args ShardedDeviceArray slow path. (jax-ml#2562)

  skye

  authored
  2e87674

  Copy full SHA for 2e87674

• Remove unused argument from pxla.shard_args (jax-ml#2560)

  skye

  authored
  e3a9a56

  Copy full SHA for e3a9a56

Commits on Mar 31, 2020

• Add ShardedDeviceArray indexing benchmark. (jax-ml#2549)

  Show description for c28c46e
  skye

  authored
  c28c46e

  Copy full SHA for c28c46e

Commits on Mar 27, 2020

• Fix pmap_benchmark.py import (jax-ml#2524)

  skye

  authored
  24bbd2b

  Copy full SHA for 24bbd2b

Commits on Mar 17, 2020

• Add pmap_benchmark.py (jax-ml#2409)

  Show description for 75077a1
  skye

  authored
  75077a1

  Copy full SHA for 75077a1

Commits on Mar 12, 2020

• jax.lax.nextafter test fix. (jax-ml#2408)

  Show description for 58feed2
  skye

  authored
  58feed2

  Copy full SHA for 58feed2

Commits on Mar 6, 2020

• Rename "NeurIPS 2019 JAX demo" to "JAX demo", as it's been presented elsewhere. (jax-ml#2372)

  skye

  authored
  74faef1

  Copy full SHA for 74faef1

• [docs] Add docstring for jax.lax.tie_in (jax-ml#2364)

  skye

  authored
  efa0315

  Copy full SHA for efa0315

Commits on Mar 5, 2020

• Document jax.device_put (jax-ml#2366)

  skye

  authored
  a1fa629

  Copy full SHA for a1fa629

Commits on Feb 23, 2020

• Shrink nested pmap image in demo notebook. (jax-ml#2293)

  skye

  authored
  f6d4837

  Copy full SHA for f6d4837

• Add nested pmap image to demo notebook. (jax-ml#2292)

  skye

  authored
  7885b1d

  Copy full SHA for 7885b1d

Commits on Feb 20, 2020

• Update WORKSPACE to pick up TF grpc upgrade and cleanup.

  skye

  committed
  0a78c8c

  Copy full SHA for 0a78c8c

Commits on Feb 19, 2020

• Update WORKSPACE to setup upstream dependencies. (jax-ml#2267)

  Show description for cfb5666
  skye

  authored
  cfb5666

  Copy full SHA for cfb5666

Commits on Feb 15, 2020

• _scatter_jvp bug fix (jax-ml#2231)

  skye

  authored
  1842093

  Copy full SHA for 1842093

Commits on Feb 14, 2020

• Bump jaxlib version in README to 0.1.39

  skye

  committed
  2a6e60f

  Copy full SHA for 2a6e60f

• Update WORKSPACE

  skye

  committed
  2ff7019

  Copy full SHA for 2ff7019

Pagination

Previous

Next