Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 29, 2023

• [JAX] Implement importing external dlpack-aware Python arrays.

  Show description for ecee8f9
  

  skye

  authored and

  jax authors

  committed
  ecee8f9

• Remove tests for jax.numpy.in1d, which is deprecated.

  Show description for e369445
  

  hawkinsp

  authored and

  jax authors

  committed
  e369445

• Remove jax.interpreters.xla.register_collective_primitive.

  Show description for 9390024
  

  hawkinsp

  authored and

  jax authors

  committed
  9390024

• Merge pull request jax-ml#17351 from jakevdp:mypy-fix

  Show description for 289ccad
  

  jax authors

  committed
  289ccad

• Add segment_ids support to pallas flash attention on GPU.

  Show description for 5a578cb
  

  wang12tao

  authored and

  jax authors

  committed
  5a578cb

• Fix mypy error

  jakevdp

  committed
  f1fc2ad

• Any devices passed to jax.sharding.Mesh are required to be hashable.

  Show description for 6072d59
  

  yashk2810

  authored and

  jax authors

  committed
  6072d59

• Merge pull request jax-ml#17337 from jakevdp:jax-extend-doc

  Show description for ff5b480
  

  jax authors

  committed
  ff5b480

• DOC: add missing docs for jax.random functions

  jakevdp

  committed
  f935c00

• Make mlir.custom_call() more general and expose it as jax.interpreters.mlir.custom_call().

  Show description for d0a6813
  

  hawkinsp

  authored and

  jax authors

  committed
  d0a6813

• Update XLA dependency to use revision

  Show description for ba8c4c3
  

  jax authors

  committed
  ba8c4c3

• Use the default CSR matmul algorithm.

  Show description for 46ac9e2
  

  hawkinsp

  authored and

  jax authors

  committed
  46ac9e2

• [Mosaic] Add missing headers in linalg vectorization.

  Show description for 046bcc0
  

  bythew3i

  authored and

  jax authors

  committed
  046bcc0

Commits on Aug 28, 2023

• Remove backward compatibility code related to pytree registries.

  Show description for 92128d4
  

  hawkinsp

  authored and

  jax authors

  committed
  92128d4

• Merge pull request jax-ml#17334 from google:cache_log

  Show description for 5bdbda5
  

  jax authors

  committed
  5bdbda5

• Don't drop out of C++ fast path if mesh pointers are not equal.

  Show description for a37e215
  

  yashk2810

  authored and

  jax authors

  committed
  a37e215

• Merge pull request jax-ml#17228 from jakevdp:doc-nb-exec

  Show description for 0c7ce72
  

  jax authors

  committed
  0c7ce72

• Merge pull request jax-ml#17331 from jakevdp:print-docs

  Show description for fbadec2
  

  jax authors

  committed
  fbadec2

• Align dummy pointers passed to cusparse to 16 bytes

  Show description for 34010a9
  

  hawkinsp

  authored and

  jax authors

  committed
  34010a9

• Turn down jax_persistent_cache_min_compile_time_secs logging from info to debug.

  Show description for c523e3b
  skye

  committed
  c523e3b

• Merge pull request jax-ml#17300 from jakevdp:prng-test

  Show description for b09bef7
  

  jax authors

  committed
  b09bef7

• DOC: mention static values in debug.print docs

  jakevdp

  committed
  2f62111

• Tests: set jax_legacy_prng_key='error'

  jakevdp

  committed
  2f878a7

• [Mosaic] Fix assert

  Show description for c3e624a
  

  tlongeri

  authored and

  jax authors

  committed
  c3e624a

• Give jax.numpy.array the type Callable.

  Show description for 70206ee
  

  hawkinsp

  authored and

  jax authors

  committed
  70206ee

• Merge pull request jax-ml#17113 from jakevdp:faster-ufuncs

  Show description for 3ea141d
  

  jax authors

  committed
  3ea141d

• jnp.ufunc: add fast paths for add/prod reductions

  jakevdp

  committed
  cb7c7ad

• Merge pull request jax-ml#17322 from abdulasiraj:patch-1

  Show description for f407298
  

  jax authors

  committed
  f407298

• Update XLA dependency to use revision

  Show description for 7b802dc
  

  jax authors

  committed
  7b802dc

• Update __init__.py

  abdulasiraj

  authored
  599b35e

• Update __init__.py to include dlpack module

  Show description for d09c55a
  abdulasiraj

  authored
  d09c55a

Commits on Aug 27, 2023

• Merge pull request jax-ml#17307 from froystig:wrap-key

  Show description for 871c9f4
  

  jax authors

  committed
  871c9f4

• Update XLA dependency to use revision

  Show description for 66c0400
  

  jax authors

  committed
  66c0400

Commits on Aug 26, 2023

• add jax.extend.random.wrap_key_data

  froystig

  committed
  a69f134

• Adds Pallas flash attention TPU kernel. Implementation based on https://arxiv.org/pdf/2205.14135.pdf.

  Show description for 841baab
  

  jax authors

  committed
  841baab

Pagination

Previous

Next