Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 5, 2023

• Accelerate deprecation of jax.ShapedArray

  Show description for c10cb17
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  c10cb17

• Fix tolerance and shard_count for experimental_rnn_test

  Show description for 8c8f50f
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  8c8f50f

• Merge pull request jax-ml#15401 from mattjj:issue15400

  Show description for f6da71c
  

  jax authors

  committed
  f6da71c

• Report the argument path when encountering an overflow error for a Python value.

  Show description for 29ba2ca
  

  hawkinsp

  authored and

  jax authors

  committed
  29ba2ca

• Merge pull request jax-ml#15415 from jakevdp:sparse-add

  Show description for 7b4e579
  

  jax authors

  committed
  7b4e579

• [sparse] allow sparse-dense add when the output is the same size as dense input

  jakevdp

  committed
  05f32a7

• Update unit test and doc how work around jit_compile=False on TPU for native_serialization.

  Show description for 0e549ac
  

  maxwillzq

  authored and

  jax authors

  committed
  0e549ac

• fix conj transpose on symbolic zero

  Show description for ac4942d
  mattjj

  committed
  ac4942d

Commits on Apr 4, 2023

• Explicitly import jax.custom_{batching,derivatives,transpose}.

  Show description for bf50551
  

  hawkinsp

  authored and

  jax authors

  committed
  bf50551

• Merge pull request jax-ml#15396 from jakevdp:conv-elem-type

  Show description for aab24fe
  

  jax authors

  committed
  aab24fe

• future-proof lax.convert_element_type

  Show description for c2fe350
  jakevdp

  committed
  c2fe350

• DCE as early as possible so that committed is not dependent on DCE's vars

  Show description for ffa9d01
  

  yashk2810

  authored and

  jax authors

  committed
  ffa9d01

• Remove PyBuffer type and its bindings.

  Show description for 9095faa
  

  pschuh

  authored and

  jax authors

  committed
  9095faa

• stages should not eagerly load the executables by calling cpp_call.

  Show description for 6040580
  

  pschuh

  authored and

  jax authors

  committed
  6040580

• Add cupti pip dependency, needed for GPU profiling.

  Show description for 75d0f65
  

  hawkinsp

  authored and

  jax authors

  committed
  75d0f65

• Avoid imports from the public jax.* namespace in more places internally.

  Show description for c1f65fc
  

  hawkinsp

  authored and

  jax authors

  committed
  c1f65fc

• Merge pull request jax-ml#15149 from sharadmv:runstate

  Show description for 3c1f3ab
  

  jax authors

  committed
  3c1f3ab

• Merge pull request jax-ml#15086 from shoyer:extrap

  Show description for efcd85d
  

  jax authors

  committed
  efcd85d

• Merge pull request jax-ml#15390 from jakevdp:checkify-dynamic-slice

  Show description for 9bb3d86
  

  jax authors

  committed
  9bb3d86

• checkify: catch OOB errors in dynamic_slice

  Show description for 46297dc
  jakevdp

  committed
  46297dc

• [shape_poly] Add some support for shape polymorphism for FFT, and tests

  Show description for 35bfdc6
  

  gnecula

  authored and

  jax authors

  committed
  35bfdc6

• Add initial implementation of a run_state primitive

  sharadmv

  committed
  5101184

• Merge pull request jax-ml#15289 from cgarciae:add-missing-api-references

  Show description for 8a6c929
  

  jax authors

  committed
  8a6c929

• Merge pull request jax-ml#15169 from cgarciae:fix-lstm

  Show description for b361f4c
  

  jax authors

  committed
  b361f4c

• Remove _compile_replicated option from compile since it is not needed anymore and some other cosmetic fixes.

  Show description for 14b572f
  

  yashk2810

  authored and

  jax authors

  committed
  14b572f

Commits on Apr 3, 2023

• Support extrapolation in jnp.interp

  Show description for 4009005
  shoyer

  committed
  4009005

• handle seq_lengths in lstm_ref

  cgarciae

  committed
  aa12e35

• Break out aot_test from array_test (for serialization and other aot APIs).

  Show description for c2b15a1
  

  pschuh

  authored and

  jax authors

  committed
  c2b15a1

• Rename count_pjit_cache_miss with count_pjit_cpp_cache_miss because it is confusing which cache the first function is taking about as pjit has many caches

  Show description for 78678ee
  

  yashk2810

  authored and

  jax authors

  committed
  78678ee

• Improve the error message of device_indices_map when the sharding is not divisible by the shape rather than raising an opaque assertion error

  Show description for 6f2256a
  

  yashk2810

  authored and

  jax authors

  committed
  6f2256a

• [jax2tf] Add more sharding tests with shape polymorphism

  Show description for 05249ec
  

  gnecula

  authored and

  jax authors

  committed
  05249ec

• [jax2tf] Skip "graph" mode primitive tests on TPUs.

  Show description for ff313a3
  

  gnecula

  authored and

  jax authors

  committed
  ff313a3

• Convolution functions in TF, like- tf.nn.depthwise_conv2d_v2, tf.nn.conv2d_transpose_v2, tf.nn.conv2d_v2 all follow the same principal when it comes to padding(explained here- https://www.tensorflo…

  Show description for d743d23
  

  jax authors

  committed
  d743d23

• Plumbing for dynamic shapes for custom calls.

  Show description for 607c7c1
  

  gnecula

  authored and

  jax authors

  committed
  607c7c1

• Merge pull request jax-ml#15340 from gnecula:dim_vars3

  Show description for 0d32724
  

  jax authors

  committed
  0d32724

Pagination

Previous

Next