Commits

Branch selector

 master

User selector

skye

Datepicker

All time

Commit History

Commits on Sep 25, 2019

• Change jit to take Device object instead of device ordinal.

  Show description for ad03baf
  skye

  committed
  ad03baf

  Copy full SHA for ad03baf

• Disable eigh test on TPU.

  skye

  committed
  f62b9d5

  Copy full SHA for f62b9d5

Commits on Sep 5, 2019

• Update README to use jaxlib-0.1.28

  skye

  committed
  2b075f9

  Copy full SHA for 2b075f9

• Add CUDA 10.1 wheels.

  skye

  committed
  2c6f74d

  Copy full SHA for 2c6f74d

• Create shim to abstract over breaking changes between jaxlib 0.1.27 and 0.1.28.

  skye

  committed
  5e0c0c9

  Copy full SHA for 5e0c0c9

• Add nn/*.py to jax/BUILD

  skye

  committed
  73d512b

  Copy full SHA for 73d512b

• Bump jaxlib version to 0.1.28 and update WORKSPACE.

  Show description for 7dc95f1
  skye

  committed
  7dc95f1

  Copy full SHA for 7dc95f1

Commits on Sep 4, 2019

• Temporarily disable test_jit_device_assignment.

  Show description for 76156e5
  skye

  committed
  76156e5

  Copy full SHA for 76156e5

Commits on Sep 3, 2019

• Update build_mac_jaxlib.yml

  skye

  authored
  5d3982e

  Copy full SHA for 5d3982e

Commits on Aug 27, 2019

• Workaround to make XRT work with multi-backend change.

  Show description for d582095
  skye

  authored
  d582095

  Copy full SHA for d582095

Commits on Aug 26, 2019

• Add jax.devices() and friends, and add devices arg to pmap.

  Show description for ae835b7
  skye

  committed
  ae835b7

  Copy full SHA for ae835b7

• Merge pull request jax-ml#1211 from levskaya/multibackend

  Show description for 2d26ac3
  skye

  authored
  2d26ac3

  Copy full SHA for 2d26ac3

Commits on Aug 25, 2019

• Update minimum jaxlib to 0.1.26.

  skye

  authored
  dfee03c

  Copy full SHA for dfee03c

Commits on Aug 24, 2019

• Bump jaxlib version to 0.1.26 and update WORKSPACE.

  skye

  committed
  4a22f56

  Copy full SHA for 4a22f56

Commits on Aug 23, 2019

• Update build_mac_jaxlib.yml

  skye

  authored
  003eda8

  Copy full SHA for 003eda8

• Update build_mac_jaxlib.yml

  skye

  authored
  3b303a5

  Copy full SHA for 3b303a5

• Update build_mac_jaxlib.yml

  skye

  authored
  24efaa7

  Copy full SHA for 24efaa7

• Remove version restriction from opt_einsum.

  Show description for 90093b7
  skye

  committed
  90093b7

  Copy full SHA for 90093b7

• Testing out github workflows

  Show description for e025fac
  skye

  authored
  e025fac

  Copy full SHA for e025fac

Commits on Aug 21, 2019

• Update GPU memory documentation.

  Show description for c50918f
  skye

  committed
  c50918f

  Copy full SHA for c50918f

Commits on Aug 20, 2019

• Require opt_einsum version to be less than 3.0.0.

  Show description for 921096e
  skye

  committed
  921096e

  Copy full SHA for 921096e

• Update jaxlib version and XLA.

  skye

  committed
  4720776

  Copy full SHA for 4720776

Commits on Jul 30, 2019

• Add how to configure TF to GPU memory allocation doc.

  skye

  committed
  3461dff

  Copy full SHA for 3461dff

• Add docs on GPU memory allocation.

  skye

  committed
  5b9849f

  Copy full SHA for 5b9849f

Commits on Jul 19, 2019

• Adding vmap/batching support to FFT.

  Show description for e716802
  

  skye

  and

  hawkinsp

  committed
  e716802

  Copy full SHA for e716802

Commits on Jun 21, 2019

• Update references to jaxlib 0.1.21

  skye

  committed
  c4cae0c

  Copy full SHA for c4cae0c

Commits on Jun 19, 2019

• Bump jaxlib version to 0.1.21 and update WORKSPACE

  skye

  committed
  8eba0e0

  Copy full SHA for 8eba0e0

Commits on Jun 12, 2019

• Bump jaxlib version to 0.1.20 and update WORKSPACE

  skye

  committed
  619b033

  Copy full SHA for 619b033

Commits on Jun 11, 2019

• Update WORKSPACE

  skye

  committed
  3ef7693

  Copy full SHA for 3ef7693

• Bump jaxlib version and WORKSPACE

  skye

  committed
  075db4a

  Copy full SHA for 075db4a

Commits on Jun 7, 2019

• pip wheels need to include the versionso use version '0.1.18' instead of 'latest'.

  skye

  committed
  04b0a91

  Copy full SHA for 04b0a91

Commits on Jun 5, 2019

• Update jaxlib version

  skye

  committed
  a61d5f6

  Copy full SHA for a61d5f6

• Update jaxlib references to 'latest'

  skye

  committed
  3a23464

  Copy full SHA for 3a23464

Commits on Jun 4, 2019

• Update Dockerfile to pip install future for py2 build.

  Show description for 3819eac
  skye

  committed
  3819eac

  Copy full SHA for 3819eac

Commits on Jun 3, 2019

• Update XLA workspace

  skye

  committed
  e071d3f

  Copy full SHA for e071d3f

Pagination

Previous

Next