Commits

Branch selector

 main

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Apr 14, 2022

• Improve precision of chlo.sinh.

  Show description for c2fe97a
  

  hawkinsp

  authored and

  jax authors

  committed
  c2fe97a

  Copy full SHA for c2fe97a

• Avoid broadcasting the input and indices in jnp.take_along_axis.

  Show description for c8ac813
  hawkinsp

  committed
  c8ac813

  Copy full SHA for c8ac813

• Fix incorrect cross-reference breaking readthedocs build.

  hawkinsp

  committed
  3f1032c

  Copy full SHA for 3f1032c

• [MHLO] Add MHLO lowerings for triangular_solve, cholesky, and schur.

  Show description for 6c1461b
  

  hawkinsp

  authored and

  jax authors

  committed
  6c1461b

  Copy full SHA for 6c1461b

• [MHLO] Add MHLO lowerings for FFT ops.

  Show description for 4806c29
  

  hawkinsp

  authored and

  jax authors

  committed
  4806c29

  Copy full SHA for 4806c29

• Disable remote_transfer_test on GPU. It currently crashes.

  Show description for b3a62cd
  

  hawkinsp

  authored and

  jax authors

  committed
  b3a62cd

  Copy full SHA for b3a62cd

• [MHLO] Add MHLO lowerings for name_p and unreachable_p.

  Show description for 8980bc4
  

  hawkinsp

  authored and

  jax authors

  committed
  8980bc4

  Copy full SHA for 8980bc4

• [MHLO] Add an MHLO lowering for rng_bit_generator.

  Show description for 665df8d
  

  hawkinsp

  authored and

  jax authors

  committed
  665df8d

  Copy full SHA for 665df8d

Commits on Apr 13, 2022

• Remove use of xla.lower_fun in SVD translation rule.

  Show description for 21f95d5
  

  hawkinsp

  authored and

  jax authors

  committed
  21f95d5

  Copy full SHA for 21f95d5

• Add libstdc++ workaround for conda users.

  hawkinsp

  committed
  2a68e7c

  Copy full SHA for 2a68e7c

• Remove mention of six package from jaxlib build instructions.

  Show description for af733ee
  hawkinsp

  committed
  af733ee

  Copy full SHA for af733ee

• Drop dead code now that the minimum jaxlib version is 0.3.2.

  hawkinsp

  committed
  94efc90

  Copy full SHA for 94efc90

• [MHLO] Separate registrations for collective and initial_style primitives from the XLA translation rule registration.

  Show description for cb4abe7
  

  hawkinsp

  authored and

  jax authors

  committed
  cb4abe7

  Copy full SHA for cb4abe7

• [MHLO] Change jax.xla_computation() to use MHLO lowering internally.

  Show description for ad8e6ad
  

  hawkinsp

  authored and

  jax authors

  committed
  ad8e6ad

  Copy full SHA for ad8e6ad

Commits on Apr 12, 2022

• [MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.

  Show description for 9455254
  

  hawkinsp

  authored and

  jax authors

  committed
  9455254

  Copy full SHA for 9455254

Commits on Apr 11, 2022

• [MHLO] Switch sharded_jit dispatch path to use MHLO lowering.

  Show description for b051be4
  

  hawkinsp

  authored and

  jax authors

  committed
  b051be4

  Copy full SHA for b051be4

Commits on Apr 8, 2022

• Revert: [MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.

  Show description for 94307a0
  

  hawkinsp

  authored and

  jax authors

  committed
  94307a0

  Copy full SHA for 94307a0

• [MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.

  Show description for 0f15fa3
  

  hawkinsp

  authored and

  jax authors

  committed
  0f15fa3

  Copy full SHA for 0f15fa3

• Revert: Pin a newer @platforms in the Bazel workspace to fix Mac ARM builds.

  Show description for 8b6b736
  

  hawkinsp

  authored and

  jax authors

  committed
  8b6b736

  Copy full SHA for 8b6b736

• [MHLO] Add direct MHLO lowerings for sparse primitives.

  Show description for 648a512
  

  hawkinsp

  authored and

  jax authors

  committed
  648a512

  Copy full SHA for 648a512

Commits on Apr 7, 2022

• Update version numbers after jax/jaxlib release.

  hawkinsp

  committed
  4dc6903

  Copy full SHA for 4dc6903

• Update libtpu version for jax 0.3.5 release.

  hawkinsp

  committed
  7f751c5

  Copy full SHA for 7f751c5

• [MHLO] Add MHLO lowerings for parallel collectives.

  Show description for cbdcdf7
  

  hawkinsp

  authored and

  jax authors

  committed
  cbdcdf7

  Copy full SHA for cbdcdf7

Commits on Apr 6, 2022

• Jax 0.3.5 and jaxlib 0.3.5 release.

  hawkinsp

  committed
  96ba290

  Copy full SHA for 96ba290

• [MHLO] Add direct MHLO lowerings for most linear algebra kernels.

  Show description for bc658e7
  

  hawkinsp

  authored and

  jax authors

  committed
  bc658e7

  Copy full SHA for bc658e7

• [MHLO] Add MHLO lowering for PRNG kernels.

  Show description for 3bfa6af
  

  hawkinsp

  authored and

  jax authors

  committed
  3bfa6af

  Copy full SHA for 3bfa6af

• [MHLO] Prefer backend-specific HLO lowerings instead of non-backend-specific MHLO lowerings.

  Show description for b9bb613
  

  hawkinsp

  authored and

  jax authors

  committed
  b9bb613

  Copy full SHA for b9bb613

• Revert: implement jnp.trace in terms of jnp.diagonal

  Show description for 4012267
  

  hawkinsp

  authored and

  jax authors

  committed
  4012267

  Copy full SHA for 4012267

• Pin a newer @platforms in the Bazel workspace to fix Mac ARM builds.

  hawkinsp

  committed
  d073a0f

  Copy full SHA for d073a0f

Commits on Apr 5, 2022

• [MHLO] Implement return type inference for GetTupleElementOp and TupleOp.

  Show description for 152c210
  

  hawkinsp

  authored and

  jax authors

  committed
  152c210

  Copy full SHA for 152c210

Commits on Apr 4, 2022

• [GPU] Force an input buffer copy for double precision complex-to-real IRFFTs.

  Show description for 71a5eb2
  

  hawkinsp

  authored and

  jax authors

  committed
  71a5eb2

  Copy full SHA for 71a5eb2

• Remove the jax_enable_mlir flag. MLIR is now the only supported code path.

  Show description for 1b8be90
  

  hawkinsp

  authored and

  jax authors

  committed
  1b8be90

  Copy full SHA for 1b8be90

Commits on Apr 1, 2022

• Avoid retracing when a host_callback.call is called multiple times with the same function.

  Show description for 208e83c
  hawkinsp

  committed
  208e83c

  Copy full SHA for 208e83c

Commits on Mar 30, 2022

• Share compare_mhlo function between lax.py and mlir.py.

  Show description for ade9f1a
  hawkinsp

  committed
  ade9f1a

  Copy full SHA for ade9f1a

Commits on Mar 29, 2022

• Add note about enabling developer mode on Windows.

  Show description for 150235c
  hawkinsp

  committed
  150235c

  Copy full SHA for 150235c

Pagination

Previous

Next