Commits

Breadcrumbs

History for

onmaster

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 18, 2020

• Minimal changes to make Jax pass a pytype check. (jax-ml#2024)

  hawkinsp

  authored
  7dbc8dc

  Copy full SHA for 7dbc8dc

Commits on Jan 15, 2020

• Add betainc to JAX (jax-ml#1998)

  Show description for 80b35dd
  

  srvasude

  authored and

  hawkinsp

  committed
  80b35dd

  Copy full SHA for 80b35dd

Commits on Jan 10, 2020

• Disable failing GPU test for now pending XLA fix.

  skye

  committed
  160cc43

  Copy full SHA for 160cc43

• Adjust tolerance for LaxTest.testConv0DIsDot. (jax-ml#1978)

  Show description for 773ebe1
  skye

  authored
  773ebe1

  Copy full SHA for 773ebe1

Commits on Jan 9, 2020

• Handle 0D convolutions correctly in shape rule. (jax-ml#1972)

  hawkinsp

  authored
  facbe0d

  Copy full SHA for facbe0d

• Implement np.sign for unsigned integers. (jax-ml#1970)

  Show description for ab25825
  hawkinsp

  authored
  ab25825

  Copy full SHA for ab25825

Commits on Jan 8, 2020

• Drop Python 2 support from JAX. (jax-ml#1962)

  Show description for dcc882c
  hawkinsp

  authored
  dcc882c

  Copy full SHA for dcc882c

• implement lazy sublanguage

  Show description for ad9b6d4
  mattjj

  committed
  ad9b6d4

  Copy full SHA for ad9b6d4

Commits on Dec 20, 2019

• Fix incorrect symbolic zero instantiation in scatter JVP rule. (jax-ml#1903)

  hawkinsp

  authored
  574a9ed

  Copy full SHA for 574a9ed

• Fix type problem in dynamic_slice_in_dim in int32 default dtype mode. (jax-ml#1902)

  hawkinsp

  authored
  178c0d8

  Copy full SHA for 178c0d8

Commits on Dec 18, 2019

• Relax fp16 test tolerance for scatter_add batching test. (jax-ml#1887)

  hawkinsp

  authored
  7df124a

  Copy full SHA for 7df124a

• Implement missing case in scatter batching rule. (jax-ml#1885)

  Show description for d692965
  hawkinsp

  authored
  d692965

  Copy full SHA for d692965

Commits on Dec 17, 2019

• Support dilated transposed convolutions in the conv_transpose op. (jax-ml#1823)

  Show description for 4af04ce
  

  tamaranorman

  authored and

  jekbradbury

  committed
  4af04ce

  Copy full SHA for 4af04ce

• Implement bool_ support for jnp.add, jnp.multiply, jnp.einsum, lax.do… (jax-ml#1872)

  Show description for b26a12a
  hawkinsp

  authored
  b26a12a

  Copy full SHA for b26a12a

Commits on Dec 11, 2019

• Implement jax.numpy.nextafter. (jax-ml#1845)

  hawkinsp

  authored
  3a07c69

  Copy full SHA for 3a07c69

Commits on Dec 6, 2019

• Fixes to type handling. (jax-ml#1824)

  Show description for fb79d56
  hawkinsp

  authored
  fb79d56

  Copy full SHA for fb79d56

Commits on Dec 2, 2019

• Relax test tolerances to fix test flakiness.

  hawkinsp

  committed
  8782860

  Copy full SHA for 8782860

• Fix bugs in handling of convolutions whose LHS has spatial size 0. (jax-ml#1794)

  Show description for f3c8af4
  hawkinsp

  authored
  f3c8af4

  Copy full SHA for f3c8af4

Commits on Nov 28, 2019

• Cleaned some test warnings.

  Show description for a47f365
  gnecula

  committed
  a47f365

  Copy full SHA for a47f365

Commits on Nov 22, 2019

• Relax test tolerances to reduce flakiness. (jax-ml#1751)

  Show description for a9d1b77
  hawkinsp

  authored
  a9d1b77

  Copy full SHA for a9d1b77

• Disable failing test (jax-ml#1744)

  skye

  authored
  3d1d140

  Copy full SHA for 3d1d140

Commits on Nov 21, 2019

• Implement batching rule for lax._select_and_gather_add (jax-ml#1736)

  hawkinsp

  authored
  a8a19e1

  Copy full SHA for a8a19e1

• Add bfloat16 support to JAX. (jax-ml#1720)

  Show description for ee36818
  hawkinsp

  authored
  ee36818

  Copy full SHA for ee36818

Commits on Nov 18, 2019

• Change scalar promotion rules to prefer array types over scalar types. (jax-ml#1709)

  Show description for 42dd736
  hawkinsp

  authored
  42dd736

  Copy full SHA for 42dd736

Commits on Nov 16, 2019

• Change test tolerance logic not to choose tolerance values based on f… (jax-ml#1701)

  Show description for bbf8129
  hawkinsp

  authored
  bbf8129

  Copy full SHA for bbf8129

Commits on Nov 15, 2019

• Move internal type-related functions into a new (internal) jax.types … (jax-ml#1695)

  Show description for f4aa515
  hawkinsp

  authored
  f4aa515

  Copy full SHA for f4aa515

• skip some tests on tpu

  mattjj

  committed
  be28700

  Copy full SHA for be28700

Commits on Nov 14, 2019

• Remove test_util.check_raises_regexp. (jax-ml#1692)

  Show description for cc0568e
  hawkinsp

  authored
  cc0568e

  Copy full SHA for cc0568e

• Add stricter type checks for the start_indices arguments to dynamic_slice and dynamic_update_slice. (jax-ml#1691)

  hawkinsp

  authored
  e670bd1

  Copy full SHA for e670bd1

Commits on Nov 11, 2019

• Faster test collection, second try (jax-ml#1653)

  Show description for a9a6cf8
  shoyer

  authored
  a9a6cf8

  Copy full SHA for a9a6cf8

Commits on Oct 22, 2019

• Add tests for float16 support in lax_test.py. (jax-ml#1553)

  Show description for 0d667d2
  hawkinsp

  authored
  0d667d2

  Copy full SHA for 0d667d2

Commits on Oct 21, 2019

• Fix lax_reference implementations of erf_inv, conj, complex. (jax-ml#1543)

  Show description for 4732ef2
  hawkinsp

  authored
  4732ef2

  Copy full SHA for 4732ef2

• Add some @jit decorators to non-primitive lax functions. (jax-ml#1542)

  Show description for abe6990
  hawkinsp

  authored
  abe6990

  Copy full SHA for abe6990

Commits on Oct 10, 2019

• add test

  jekbradbury

  committed
  9d2f25c

  Copy full SHA for 9d2f25c

Commits on Sep 15, 2019

• add some 'manual' lax.rem autodiff tests

  mattjj

  committed
  e945c9c

  Copy full SHA for e945c9c

Pagination

Previous

Next