Commits

Branch selector

 main

User selector

jakevdp

All time

Commit History

Commits on Aug 17, 2024

• Improve type annotations for jax.Array methods

  jakevdp

  committed
  dd697a9

  Copy full SHA for dd697a9

Commits on Aug 16, 2024

• document jax.Array methods and attributes

  jakevdp

  committed
  e9d6fd3

  Copy full SHA for e9d6fd3

Commits on Aug 14, 2024

• Document methods of custom_jvp/custom_vjp

  jakevdp

  committed
  91f5512

  Copy full SHA for 91f5512

• Deprecate several internal utilities in jax.core

  jakevdp

  committed
  bd9698e

  Copy full SHA for bd9698e

• Use PEP484-style exports in several submodules

  jakevdp

  committed
  5cc6899

  Copy full SHA for 5cc6899

• Add method argument for jnp.isin

  jakevdp

  committed
  25da7ad

  Copy full SHA for 25da7ad

Commits on Aug 13, 2024

• doc: clarify data_fields & meta_fields in register_dataclass

  jakevdp

  committed
  5903c77

  Copy full SHA for 5903c77

• jnp.setxor1d: add support for static size argument

  jakevdp

  committed
  52c269c

  Copy full SHA for 52c269c

Commits on Aug 12, 2024

• test: improve test cases for set-like operations

  Show description for 26800f1
  jakevdp

  committed
  26800f1

  Copy full SHA for 26800f1

Commits on Aug 10, 2024

• jnp.intersect1d: add support for static size argument.

  jakevdp

  committed
  c2c116d

  Copy full SHA for c2c116d

Commits on Aug 9, 2024

• Better docs for array, asarray, linspace

  Show description for 8bd8491
  jakevdp

  committed
  8bd8491

  Copy full SHA for 8bd8491

Commits on Aug 8, 2024

• [array API] update test suite to most recent commit

  jakevdp

  committed
  d999208

  Copy full SHA for d999208

• Improve documentation for jnp.piecewise & jnp.select

  jakevdp

  committed
  4ca3417

  Copy full SHA for 4ca3417

Commits on Aug 7, 2024

• CI: fix mypy errors

  jakevdp

  committed
  53af0d4

  Copy full SHA for 53af0d4

Commits on Aug 6, 2024

• deprecate jax.lib.xla_client.bfloat16

  jakevdp

  committed
  a009e1c

  Copy full SHA for a009e1c

• Support empty boolean indexing

  jakevdp

  committed
  b45f0fe

  Copy full SHA for b45f0fe

• CI: pin sphinx to avoid build errors on 8.0

  jakevdp

  committed
  4f8c5a3

  Copy full SHA for 4f8c5a3

Commits on Aug 5, 2024

• Deprecate jax.lib.xla_client._xla

  Show description for 06f29bb
  jakevdp

  committed
  06f29bb

  Copy full SHA for 06f29bb

Commits on Aug 3, 2024

• Improve documentation for jnp.from_dlpack

  jakevdp

  committed
  4d637c8

  Copy full SHA for 4d637c8

• Tighten the public API for jax.lib.xla_client & xla_extension

  jakevdp

  committed
  521c94c

  Copy full SHA for 521c94c

Commits on Aug 2, 2024

• remove jax.extend.backend.default_backend in favor of jax.backend

  Show description for 3fa86a9
  jakevdp

  committed
  3fa86a9

  Copy full SHA for 3fa86a9

Commits on Aug 1, 2024

• [array api] fix deprecation to support old import pattern

  jakevdp

  committed
  48c5fab

  Copy full SHA for 48c5fab

• [array api] Finalize array API in jax.numpy & deprecate jax.experimental.array_api

  jakevdp

  committed
  14fa062

  Copy full SHA for 14fa062

Commits on Jul 31, 2024

• jnp.where: explicitly use mode='drop' when out-of-bound indices expected

  jakevdp

  committed
  a9690ad

  Copy full SHA for a9690ad

• Improve error when indexing with too many indices

  jakevdp

  committed
  9a88ecb

  Copy full SHA for 9a88ecb

• [array api] use jnp.astype directly

  jakevdp

  committed
  c24f209

  Copy full SHA for c24f209

Commits on Jul 30, 2024

• jnp.repeat: add copy argument for Array API

  jakevdp

  committed
  5198db9

  Copy full SHA for 5198db9

• [array api] remove redundant definitions for clip() & hypot()

  jakevdp

  committed
  1259322

  Copy full SHA for 1259322

• [array API] move api metadata into jax.numpy namespace

  jakevdp

  committed
  c2f2b0e

  Copy full SHA for c2f2b0e

Commits on Jul 29, 2024

• [array API] add device argument to fftfreq & rfftfreq

  jakevdp

  committed
  6516a07

  Copy full SHA for 6516a07

• [array API] move api metadata into jax.numpy namespace

  jakevdp

  committed
  00ba7a6

  Copy full SHA for 00ba7a6

• Improve documentation for jnp.put, jnp.place, jnp.fill_diagonal

  Show description for cfa1e78
  jakevdp

  committed
  cfa1e78

  Copy full SHA for cfa1e78

Commits on Jul 25, 2024

• [array api] streamline astype device implementation

  Show description for 81b9db6
  jakevdp

  committed
  81b9db6

  Copy full SHA for 81b9db6

• [array API] clean up some superseded definitions

  jakevdp

  committed
  8c36f90

  Copy full SHA for 8c36f90

Commits on Jul 24, 2024

• [array API] clean up some unused/unnecessary code

  jakevdp

  committed
  b8bb869

  Copy full SHA for b8bb869