Commits

Branch selector

 main

User selector

hawkinsp

All time

Commit History

Commits on Dec 15, 2023

• Add a warning if the user calls os.fork().

  Show description for ec89e5e
  hawkinsp

  committed
  ec89e5e

  Copy full SHA for ec89e5e

Commits on Dec 14, 2023

• Recommend the plugin in the CUDA installation instructions.

  hawkinsp

  committed
  78bc4f5

  Copy full SHA for 78bc4f5

• Remove XLA logging patch, now the XLA fix has landed.

  Show description for cf5a495
  

  hawkinsp

  authored and

  jax authors

  committed
  cf5a495

  Copy full SHA for cf5a495

• Add patch to suppress XLA:GPU logging.

  Show description for b392622
  

  hawkinsp

  authored and

  jax authors

  committed
  b392622

  Copy full SHA for b392622

Commits on Dec 11, 2023

• [XLA:Python] Add support for explicitly creating the gloo tcp context.

  Show description for 384e29e
  

  hawkinsp

  authored and

  jax authors

  committed
  384e29e

  Copy full SHA for 384e29e

Commits on Dec 6, 2023

• Add register_jax_dialects to jaxlib wheel.

  Show description for 5601873
  hawkinsp

  committed
  5601873

  Copy full SHA for 5601873

• Add jax.extend.mlir.

  Show description for 78543f7
  

  hawkinsp

  authored and

  jax authors

  committed
  78543f7

  Copy full SHA for 78543f7

• Remove stale reference to _site_initialize_0 in wheel build script.

  hawkinsp

  committed
  1c80b36

  Copy full SHA for 1c80b36

• Use an explicit MLIR dialect registration, rather than _site_initialize_0.

  Show description for d95084d
  

  hawkinsp

  authored and

  jax authors

  committed
  d95084d

  Copy full SHA for d95084d

Commits on Dec 5, 2023

• [bazel] Add a macro if_building_jaxlib() to guard dependencies that should only be present if building jaxlib.

  Show description for 720ff42
  

  hawkinsp

  authored and

  jax authors

  committed
  720ff42

  Copy full SHA for 720ff42

• [bazel] Add a BUILD file for jax/extend, and add more granular targets for individual pieces of extend.

  Show description for 7fa0f46
  

  hawkinsp

  authored and

  jax authors

  committed
  7fa0f46

  Copy full SHA for 7fa0f46

Commits on Dec 4, 2023

• Use sys.version_info to guard Python 3.11 only code.

  Show description for 193eb12
  

  hawkinsp

  authored and

  jax authors

  committed
  193eb12

  Copy full SHA for 193eb12

• Use scoped disable_jit() in dynamic_api_test.

  Show description for baa7756
  

  hawkinsp

  authored and

  jax authors

  committed
  baa7756

  Copy full SHA for baa7756

• Disable sanitizer builds of shape_poly_test.

  Show description for 5b97960
  

  hawkinsp

  authored and

  jax authors

  committed
  5b97960

  Copy full SHA for 5b97960

• Disable export_harnesses_multi_platform_test under sanitizers.

  Show description for 1d95e79
  

  hawkinsp

  authored and

  jax authors

  committed
  1d95e79

  Copy full SHA for 1d95e79

Commits on Dec 2, 2023

• Remove the ml_program MLIR dialect from jaxlib.

  Show description for 32fb1b4
  

  hawkinsp

  authored and

  jax authors

  committed
  32fb1b4

  Copy full SHA for 32fb1b4

Commits on Dec 1, 2023

• Remove nightly NVIDIA GPU multiprocess CI.

  Show description for 7274960
  hawkinsp

  committed
  7274960

  Copy full SHA for 7274960

• Improve error message when cudnn is not found.

  Show description for a999120
  

  hawkinsp

  authored and

  jax authors

  committed
  a999120

  Copy full SHA for a999120

• Fix Windows build failure.

  Show description for 50c7223
  

  hawkinsp

  authored and

  jax authors

  committed
  50c7223

  Copy full SHA for 50c7223

• Inline sigmoid, isfinite, and isnan in jaxprs.

  Show description for 95bc2ba
  

  hawkinsp

  authored and

  jax authors

  committed
  95bc2ba

  Copy full SHA for 95bc2ba

Commits on Nov 29, 2023

• Use process_count(backend) in local_devices().

  Show description for 842ca2c
  

  hawkinsp

  authored and

  jax authors

  committed
  842ca2c

  Copy full SHA for 842ca2c

• Always lower reduce_scatter_p as an HLO ReduceScatter.

  Show description for 458a896
  

  hawkinsp

  authored and

  jax authors

  committed
  458a896

  Copy full SHA for 458a896

• Remove fallback path that lowers all_gather via psum.

  Show description for 1e961b8
  

  hawkinsp

  authored and

  jax authors

  committed
  1e961b8

  Copy full SHA for 1e961b8

Commits on Nov 28, 2023

• Build CUDA kernels for sm_50 instead of sm_52.

  Show description for 6fa506c
  

  hawkinsp

  authored and

  jax authors

  committed
  6fa506c

  Copy full SHA for 6fa506c

Commits on Nov 21, 2023

• Request specific version information in the bug template.

  hawkinsp

  committed
  6a1dde6

  Copy full SHA for 6a1dde6

• Make jax.numpy.where()'s condition, x, y arguments positional-only to match numpy.where.

  Show description for 84c1e82
  

  hawkinsp

  authored and

  jax authors

  committed
  84c1e82

  Copy full SHA for 84c1e82

• Fix error/hang when non-finite values are passed to non-symmetric Eigendecomposition.

  Show description for 49c80e6
  

  hawkinsp

  authored and

  jax authors

  committed
  49c80e6

  Copy full SHA for 49c80e6

Commits on Nov 20, 2023

• Apply jupytext fix to #18220.

  hawkinsp

  committed
  be9cafc

  Copy full SHA for be9cafc

• Disable export_harnesses_test under asan.

  Show description for fd9a1a2
  

  hawkinsp

  authored and

  jax authors

  committed
  fd9a1a2

  Copy full SHA for fd9a1a2

Commits on Nov 18, 2023

• Add minimum version checks for cublas and cusparse.

  Show description for 41f0b33
  

  hawkinsp

  authored and

  jax authors

  committed
  41f0b33

  Copy full SHA for 41f0b33

Commits on Nov 17, 2023

• Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.

  Show description for 8e8dc26
  

  hawkinsp

  authored and

  jax authors

  committed
  8e8dc26

  Copy full SHA for 8e8dc26

• Increase minimum jaxlib version to 0.4.19.

  Show description for 30a0136
  

  hawkinsp

  authored and

  jax authors

  committed
  30a0136

  Copy full SHA for 30a0136

Commits on Nov 15, 2023

• Reverts ef9075159a67a2b94526b65e4a2c2904a4a49046

  Show description for 234be73
  

  hawkinsp

  authored and

  jax authors

  committed
  234be73

  Copy full SHA for 234be73

• [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

  Show description for 0560cc4
  

  hawkinsp

  authored and

  jax authors

  committed
  0560cc4

  Copy full SHA for 0560cc4

Commits on Nov 14, 2023

• [JAX:GPU] Generalize gesvdj kernel to iterate over the unbatched Jacobi kernel in cases that we cannot use the batched kernel.

  Show description for 95e2d3f
  

  hawkinsp

  authored and

  jax authors

  committed
  95e2d3f

  Copy full SHA for 95e2d3f

Pagination

Previous

Next