Commits

Breadcrumbs

History for

on2ca151e

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 20, 2023

• [XLA:Python] Fix __cuda_array_interface__.

  Show description for 0ec03db
  

  hawkinsp

  authored and

  jax authors

  committed
  0ec03db

  Copy full SHA for 0ec03db

Commits on Jun 15, 2023

• Allow None to be passed to in_shardings and out_shardings. The default is still UNSPECIFIED to handle edge cases around the old semantics where None is treated as fully replicated.

  Show description for 6007698
  

  yashk2810

  authored and

  jax authors

  committed
  6007698

  Copy full SHA for 6007698

Commits on Jun 13, 2023

• [jax2tf] Add native_lowering_disabled_checks parameter to jax2tf.convert.

  Show description for 0961fb9
  gnecula

  committed
  0961fb9

  Copy full SHA for 0961fb9

Commits on Jun 12, 2023

• Fix cuda12 pip install.

  Show description for 0ded163
  hawkinsp

  committed
  0ded163

  Copy full SHA for 0ded163

Commits on Jun 8, 2023

• Update versions and changelog post 0.4.12 release

  skye

  committed
  4b80103

  Copy full SHA for 4b80103

• Delete OpShardingSharding export since it has been 3 months since it was deprecated. Also remove deprecation warnings for MeshPspecSharding.

  Show description for 1445149
  

  yashk2810

  authored and

  jax authors

  committed
  1445149

  Copy full SHA for 1445149

• [XLA:Python] Fix crash when accessing locals of JAX-generated Python tracebacks.

  Show description for 6374b73
  

  hawkinsp

  authored and

  jax authors

  committed
  6374b73

  Copy full SHA for 6374b73

• [XLA:Python] Fix incorrect code source information under Python 3.11.

  Show description for a47aca8
  

  hawkinsp

  authored and

  jax authors

  committed
  a47aca8

  Copy full SHA for a47aca8

• Merge pull request jax-ml#16246 from chrisflesher:scipy-rotation-v3

  Show description for 8d27f20
  

  jax authors

  committed
  8d27f20

  Copy full SHA for 8d27f20

• Added scipy.spatial.transform Rotation and Slerp classes

  chrisflesher

  committed
  5be17ed

  Copy full SHA for 5be17ed

• Mark jax.abstract_arrays as deprecated

  jakevdp

  committed
  47ae5bd

  Copy full SHA for 47ae5bd

Commits on Jun 7, 2023

• [PJRT:C] Implement PJRT_Device_MemoryStats

  Show description for 1a3ac88
  

  skye

  authored and

  jax authors

  committed
  1a3ac88

  Copy full SHA for 1a3ac88

Commits on Jun 5, 2023

• Include SASS/PTX for Hopper GPUs.

  hawkinsp

  committed
  cb33fdf

  Copy full SHA for cb33fdf

Commits on Jun 2, 2023

• Deprecate jax.numpy functions alltrue, sometrue, product, cumproduct

  jakevdp

  committed
  3bef621

  Copy full SHA for 3bef621

Commits on Jun 1, 2023

• Finish jax and jaxlib 0.4.11 release

  Show description for 4c48611
  

  yashk2810

  authored and

  jax authors

  committed
  4c48611

  Copy full SHA for 4c48611

Commits on May 28, 2023

• Deprecate jax.interpreters.xla.Buffer, device_put, xla_call_p

  jakevdp

  committed
  7a87995

  Copy full SHA for 7a87995

Commits on May 26, 2023

• Remove axis_resources from with_sharding_constraint since it has been 3 months since the deprecation as per the API deprecation policy.

  Show description for fe3fed3
  

  yashk2810

  authored and

  jax authors

  committed
  fe3fed3

  Copy full SHA for fe3fed3

Commits on May 25, 2023

• Reland: [XLA:Python] Add buffer protocol support to jax.Array

  Show description for e464dc8
  

  hawkinsp

  authored and

  jax authors

  committed
  e464dc8

  Copy full SHA for e464dc8

Commits on May 24, 2023

• Copybara import of the project:

  Show description for 399e4ee
  

  jakevdp

  authored and

  jax authors

  committed
  399e4ee

  Copy full SHA for 399e4ee

Commits on May 12, 2023

• Update versions and changelog post 0.4.10 release

  skye

  committed
  533a7c0

  Copy full SHA for 533a7c0

Commits on May 11, 2023

• Update setup.py, WORKSPACE, and CHANGELOG for jax/jaxlib 0.4.10 release

  skye

  committed
  82bbeef

  Copy full SHA for 82bbeef

Commits on May 10, 2023

• Merge pull request jax-ml#15947 from skye:version

  Show description for bbc9632
  

  jax authors

  committed
  bbc9632

  Copy full SHA for bbc9632

• Add improved TPU SVD accuracy to the changelog.

  Show description for cc5e694
  

  hawkinsp

  authored and

  jax authors

  committed
  cc5e694

  Copy full SHA for cc5e694

• Update versions and changelog for 0.4.9 release

  skye

  committed
  b02b043

  Copy full SHA for b02b043

Commits on May 2, 2023

• [Rollback] Remove py3.8 support from jax as per https://jax.readthedocs.io/en/latest/deprecation.html

  Show description for 356cac0
  

  yashk2810

  authored and

  jax authors

  committed
  356cac0

  Copy full SHA for 356cac0

Commits on May 1, 2023

• Remove py3.8 support from jax as per https://jax.readthedocs.io/en/latest/deprecation.html

  Show description for e51d12c
  

  yashk2810

  authored and

  jax authors

  committed
  e51d12c

  Copy full SHA for e51d12c

Commits on Apr 27, 2023

• Revert: Switch to using Clang as the default compiler.

  Show description for 84c5169
  

  hawkinsp

  authored and

  jax authors

  committed
  84c5169

  Copy full SHA for 84c5169

Commits on Apr 25, 2023

• Switch to using Clang as the default compiler.

  Show description for 782d90d
  

  pschuh

  authored and

  jax authors

  committed
  782d90d

  Copy full SHA for 782d90d

Commits on Apr 14, 2023

• Deprecate and raise an exception for instantiate_const_outputs argument of jax.xla_computation since it has been unused for a very long time.

  Show description for 30c6871
  

  yashk2810

  authored and

  jax authors

  committed
  30c6871

  Copy full SHA for 30c6871

Commits on Apr 13, 2023

• Remove in_parts, out_parts from jax.xla_computation since they were only used for sharded_jit and sharded_jit is long gone

  Show description for 3e93833
  

  yashk2810

  authored and

  jax authors

  committed
  3e93833

  Copy full SHA for 3e93833

Commits on Apr 7, 2023

• Remove experimental_cpp_pmap flag since it is always on

  Show description for 738dd71
  

  yashk2810

  authored and

  jax authors

  committed
  738dd71

  Copy full SHA for 738dd71

• Remove experimental_cpp_jit since that flag is unused and also remove experimental_cpp_pjit.

  Show description for 694e43a
  

  yashk2810

  authored and

  jax authors

  committed
  694e43a

  Copy full SHA for 694e43a

Commits on Apr 1, 2023

• Rename gda_serialization to array_serialization but keep gda_serialization around until it is included in a jax release so that OSS projects can be moved to array_serialization

  Show description for d27a80d
  

  yashk2810

  authored and

  jax authors

  committed
  d27a80d

  Copy full SHA for d27a80d

Commits on Mar 31, 2023

• Remove deprecated function jnp.msort

  jakevdp

  committed
  749dc1b

  Copy full SHA for 749dc1b

Commits on Mar 30, 2023

• Raise deprecation warnings for {in|out}_axis_resources for pjit and axis_resources for with_sharding_constraint

  Show description for 69c9660
  

  yashk2810

  authored and

  jax authors

  committed
  69c9660

  Copy full SHA for 69c9660

Pagination

Previous

Next