Commits

Branch selector

 main

User selector

sharadmv

Datepicker

All time

Commit History

Commits on Apr 1, 2022

• Fixes jax2tf's test_name_scope to use graph introspection instead of

  Show description for aac8ec8
  sharadmv

  committed
  aac8ec8

  Copy full SHA for aac8ec8

Commits on Mar 30, 2022

• Fix link to buffer-donation docs

  sharadmv

  committed
  fb97717

  Copy full SHA for fb97717

• Remove redundant name-stack setting in DynamicJaxprTrace

  sharadmv

  committed
  c233a97

  Copy full SHA for c233a97

Commits on Mar 10, 2022

• Refactor Jaxpr pretty-printing to use a JaxprPpSettings named tuple

  Show description for 2988901
  sharadmv

  committed
  2988901

  Copy full SHA for 2988901

Commits on Feb 23, 2022

• Add separate mechanism for threading name stacks to the lowering

  sharadmv

  committed
  1b79caa

  Copy full SHA for 1b79caa

Commits on Oct 29, 2021

• Fix forward for PR jax-ml#8392 (made source_info for new_jaxpr_eqn argument optional again)

  Show description for 32319e1
  

  sharadmv

  authored and

  jax authors

  committed
  32319e1

  Copy full SHA for 32319e1

Commits on Oct 28, 2021

• Use SourceInfo named tuple to keep track of source information

  sharadmv

  committed
  b40245e

  Copy full SHA for b40245e

Commits on Oct 27, 2021

• [JAX] Update JAX users in preparation for a change that makes iteration over a JAX array return JAX arrays, instead of NumPy arrays.

  Show description for b6fa33f
  

  sharadmv

  authored and

  jax authors

  committed
  b6fa33f

  Copy full SHA for b6fa33f

• Broadcast unmapped values in all_to_all batching rule

  Show description for ae9e698
  

  sharadmv

  and

  apaszke

  committed
  ae9e698

  Copy full SHA for ae9e698

Commits on Sep 16, 2021

• Fix for singleton axis name in axis_index_translation_rule

  sharadmv

  committed
  8ae58be

  Copy full SHA for 8ae58be

Commits on Sep 13, 2021

• Add precision param for pdot

  sharadmv

  committed
  ebd8d95

  Copy full SHA for ebd8d95

Commits on Sep 9, 2021

• Combine initial_style_batchers with collective_rules

  sharadmv

  committed
  cc3e197

  Copy full SHA for cc3e197

Commits on Sep 3, 2021

• change while loop batching fixed point condition

  Show description for d693324
  

  sharadmv

  and

  apaszke

  committed
  d693324

  Copy full SHA for d693324

Commits on Aug 25, 2021

• Pass x into transpose in autodidax

  sharadmv

  committed
  0fa7008

  Copy full SHA for 0fa7008

Commits on Jul 28, 2021

• change while loop batching fixed point condition

  Show description for 49f7ac2
  

  sharadmv

  and

  apaszke

  committed
  49f7ac2

  Copy full SHA for 49f7ac2

Commits on Apr 13, 2021

• Update references in custom interpreter notebook

  sharadmv

  committed
  577c601

  Copy full SHA for 577c601

Commits on Mar 22, 2021

• Add support for custom derivatives in jax.experimental.callback

  sharadmv

  committed
  252bd6c

  Copy full SHA for 252bd6c

Commits on Mar 10, 2021

• Add scan and while rule for jax.experimental.callback transformation

  sharadmv

  committed
  ddaef19

  Copy full SHA for ddaef19

Commits on Jan 26, 2021

• Allow jax.custom_gradient to return vjp with singleton return value

  sharadmv

  committed
  6061b09

  Copy full SHA for 6061b09

Commits on Nov 18, 2020

• Add support for XLA variadic reduce

  sharadmv

  committed
  590e629

  Copy full SHA for 590e629

Commits on Nov 10, 2020

• Fix typo in FAQ

  sharadmv

  authored
  a51afa3

  Copy full SHA for a51afa3

Commits on Oct 16, 2020

• fix lint error

  sharadmv

  committed
  cae5334

  Copy full SHA for cae5334

• Add default implementation of process_custom_jvp_call and

  Show description for e8901d5
  sharadmv

  committed
  e8901d5

  Copy full SHA for e8901d5

Commits on Aug 25, 2020

• Remove frame check assertion in extend_axis_env.

  sharadmv

  committed
  774b5f6

  Copy full SHA for 774b5f6

Commits on May 13, 2020

• Make jnp.array convert empty list to DeviceArray (jax-ml#3049)

  Show description for e9d3394
  

  sharadmv

  and

  hawkinsp

  authored
  e9d3394

  Copy full SHA for e9d3394

Commits on Apr 5, 2020

• Fix lax.broadcast_shapes returning numpy ints in shape tuple (jax-ml#2471)

  Show description for 99944d1
  sharadmv

  authored
  99944d1

  Copy full SHA for 99944d1

Commits on Apr 3, 2020

• Fix dtype error

  sharadmv

  committed
  3c9fb35

  Copy full SHA for 3c9fb35

• Fix grad(logit) to use defjvps and enable it in tests

  sharadmv

  committed
  72783bb

  Copy full SHA for 72783bb

Commits on Mar 27, 2020

• Instantiate zeros in _custom_vjp_call_jaxpr_jvp

  sharadmv

  committed
  0499b84

  Copy full SHA for 0499b84

Commits on Mar 18, 2020

• Fix xlog1py and xlogy not returning 0 when x == 0. (jax-ml#2450)

  Show description for cd248ba
  sharadmv

  authored
  cd248ba

  Copy full SHA for cd248ba

Commits on Feb 15, 2020

• Set call_p.multiple_results to True.

  sharadmv

  authored
  b92656d

  Copy full SHA for b92656d

Commits on Feb 13, 2020

• Make core.call_p a call primitive. (jax-ml#2223)

  sharadmv

  authored
  e936974

  Copy full SHA for e936974

Commits on Feb 10, 2020

• Fix inconsistent indentation in JaxprTrace.default_process_primitive.

  sharadmv

  authored
  76d77bf

  Copy full SHA for 76d77bf

Commits on Dec 21, 2019

• Fix typo in scipy.stats.poisson.pmf (jax-ml#1894)

  Show description for 5b4106f
  

  sharadmv

  authored and

  jekbradbury

  committed
  5b4106f

  Copy full SHA for 5b4106f

Commits on Nov 8, 2019

• Extend clip test to test broadcasting a_min and a_max

  sharadmv

  committed
  6c305f3

  Copy full SHA for 6c305f3

Pagination

Previous

Next