Commits

Breadcrumbs

History for

on3b2410f

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 31, 2025

• [pjrt] Use the PjRtMemorySpace version of BufferFromHostLiteral

  Show description for 5cad02c
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  5cad02c

  Copy full SHA for 5cad02c

Commits on Jan 13, 2025

• Internal change.

  Show description for e72c148
  

  dfm

  authored and

  Google-ML-Automation

  committed
  e72c148

  Copy full SHA for e72c148

Commits on Jan 7, 2025

• Add an example demonstrating input-output aliasing with the FFI.

  dfm

  committed
  62656b3

  Copy full SHA for 62656b3

Commits on Dec 29, 2024

• Move jex.ffi to jax.ffi.

  dfm

  committed
  cb4d97a

  Copy full SHA for cb4d97a

Commits on Dec 12, 2024

• Don't monkey-patch functions in test_utils to count events for tests.

  Show description for 62e66b6
  hawkinsp

  committed
  62e66b6

  Copy full SHA for 62e66b6

Commits on Nov 25, 2024

• Refactor FFI examples to consolidate several examples into one submodule.

  dfm

  committed
  84a9cba

  Copy full SHA for 84a9cba

Commits on Nov 19, 2024

• Move JAX example to public XLA:CPU API

  Show description for 42fbd30
  

  changm

  authored and

  Google-ML-Automation

  committed
  42fbd30

  Copy full SHA for 42fbd30

Commits on Nov 14, 2024

• Merge pull request #23805 from dfm:ffi-examples-state

  Show description for 8370082
  Google-ML-Automation

  committed
  8370082

  Copy full SHA for 8370082

Commits on Nov 12, 2024

• Add an example FFI call to demonstrate the use of global state.

  dfm

  committed
  f086483

  Copy full SHA for f086483

• Update some outdated syntax in FFI tutorial.

  dfm

  committed
  f757054

  Copy full SHA for f757054

Commits on Oct 31, 2024

• Move the CUDA end-to-end example to FFI examples workflow + hosted

  Show description for ce8dba9
  dfm

  committed
  ce8dba9

  Copy full SHA for ce8dba9

Commits on Oct 21, 2024

• Rename vmap methods for callbacks.

  dfm

  committed
  61701af

  Copy full SHA for 61701af

• Make ffi_call return a callable

  dfm

  committed
  0b651f0

  Copy full SHA for 0b651f0

Commits on Oct 2, 2024

• Deprecate the vectorized argument to pure_callback and ffi_call.

  dfm

  committed
  1d27d42

  Copy full SHA for 1d27d42

Commits on Oct 1, 2024

• Add support for passing array attributes via ffi_call

  dfm

  committed
  f60c5cc

  Copy full SHA for f60c5cc

Commits on Sep 25, 2024

• Merge pull request #23409 from dfm:ffi-examples

  Show description for 9d277e6
  Google-ML-Automation

  committed
  9d277e6

  Copy full SHA for 9d277e6

Commits on Sep 24, 2024

• Add FFI example project and test on CI.

  Show description for e1a68ee
  dfm

  committed
  e1a68ee

  Copy full SHA for e1a68ee

Commits on Aug 26, 2024

• Clean up BUILD files.

  Show description for 6d1f51e
  

  hawkinsp

  authored and

  jax authors

  committed
  6d1f51e

  Copy full SHA for 6d1f51e

Commits on Jun 20, 2024

• Stop using xla/statusor.h now that it just contains an alias for absl::Status.

  Show description for 84d748f
  

  klucke

  authored and

  jax authors

  committed
  84d748f

  Copy full SHA for 84d748f

Commits on May 21, 2024

• Automated Code Change

  Show description for 418b688
  

  klucke

  authored and

  jax authors

  committed
  418b688

  Copy full SHA for 418b688

Commits on May 10, 2024

• Force float32 matmuls in examples_test.

  Show description for 24b4731
  hawkinsp

  committed
  24b4731

  Copy full SHA for 24b4731

Commits on May 9, 2024

• Reenable examples_test in Bazel build.

  Show description for 89d25bb
  

  hawkinsp

  authored and

  jax authors

  committed
  89d25bb

  Copy full SHA for 89d25bb

Commits on Apr 11, 2024

• Avoid 'from jax import config' imports

  Show description for f090074
  jakevdp

  committed
  f090074

  Copy full SHA for f090074

Commits on Apr 3, 2024

• [PJRT:CPU] Replace references to pjrt/tfrt_cpu_pjrt_client with pjrt/cpu/cpu_client.h.

  Show description for 1baed9b
  

  hawkinsp

  authored and

  jax authors

  committed
  1baed9b

  Copy full SHA for 1baed9b

Commits on Mar 7, 2024

• update top-level examples to use new-style typed keys

  froystig

  committed
  78fd4f1

  Copy full SHA for 78fd4f1

Commits on Jan 25, 2024

• [XLA][NFC] Make interface of module loaders consistent.

  Show description for fb80d2a
  

  olegshyshkov

  authored and

  jax authors

  committed
  fb80d2a

  Copy full SHA for fb80d2a

Commits on Aug 24, 2023

• Switch jaxlib to use nanobind instead of pybind11.

  Show description for 70b7d50
  

  hawkinsp

  authored and

  jax authors

  committed
  70b7d50

  Copy full SHA for 70b7d50

Commits on Aug 9, 2023

• Fix for log-normal prior in example.

  cottrell

  committed
  40d0d40

  Copy full SHA for 40d0d40

Commits on Jul 27, 2023

• Update flags to use the ABSL typed flag API.

  Show description for 76cda0a
  

  hawkinsp

  authored and

  jax authors

  committed
  76cda0a

  Copy full SHA for 76cda0a

Commits on Apr 21, 2023

• Change to simpler import for jax.config

  jakevdp

  committed
  fbe4f10

  Copy full SHA for fbe4f10

Commits on Apr 19, 2023

• [JAX] Improve handling of metadata in compilation cache.

  Show description for 3bb7386
  

  hawkinsp

  authored and

  jax authors

  committed
  3bb7386

  Copy full SHA for 3bb7386

Commits on Apr 10, 2023

• Make differential privacy example work again.

  Show description for 9f03bd5
  

  hawkinsp

  authored and

  jax authors

  committed
  9f03bd5

  Copy full SHA for 9f03bd5

Commits on Mar 13, 2023

• Fix build breakage from OpenXLA switch.

  Show description for ab45383
  

  hawkinsp

  authored and

  jax authors

  committed
  ab45383

  Copy full SHA for ab45383

• Merge pull request #14475 from hawkinsp:openxla

  Show description for 42ef649
  

  jax authors

  committed
  42ef649

  Copy full SHA for 42ef649

• Switch JAX to use the OpenXLA repository.

  hawkinsp

  committed
  172a831

  Copy full SHA for 172a831

Pagination

Previous

Next