Commits

Branch selector

 main

User selector

froystig

Datepicker

All time

Commit History

Commits on Mar 29, 2022

• require consistent output structure in custom vmap rules

  Show description for b2de101
  froystig

  committed
  b2de101

  Copy full SHA for b2de101

• add custom_vjp upgrade flag to jax module and fix doc rendering for upgrade flags

  froystig

  committed
  e0ddd43

  Copy full SHA for e0ddd43

Commits on Mar 28, 2022

• refine occurrences of the "forward-mode disallowed" error under new custom_vjp

  Show description for 941b221
  

  froystig

  and

  mattjj

  committed
  941b221

  Copy full SHA for 941b221

• introduce partly-working new custom_vjp implementation behind a flag

  Show description for 52a1618
  froystig

  committed
  52a1618

  Copy full SHA for 52a1618

• introduce an upgrade flag for custom_vjp overhaul (still experimental)

  Show description for 1d9a6c1
  froystig

  committed
  1d9a6c1

  Copy full SHA for 1d9a6c1

Commits on Mar 26, 2022

• allow for recursive uses of custom_transpose

  Show description for a6a43e2
  

  froystig

  and

  mattjj

  committed
  a6a43e2

  Copy full SHA for a6a43e2

Commits on Mar 24, 2022

• add jaxpr check in partial_eval.trace_to_subjaxpr_dynamic

  froystig

  committed
  d1f49c1

  Copy full SHA for d1f49c1

Commits on Mar 23, 2022

• remove unreachable code in Lowered

  froystig

  committed
  fa89321

  Copy full SHA for fa89321

• protocols for the lowering and executable underlying Lowered and Compiled

  froystig

  committed
  e317863

  Copy full SHA for e317863

• avoid batching units in cond partial eval

  Show description for 0ada0a1
  

  froystig

  and

  mattjj

  committed
  0ada0a1

  Copy full SHA for 0ada0a1

Commits on Mar 22, 2022

• fix jaxpr pretty-printing with unitvars

  Show description for 090f6e5
  

  froystig

  and

  mattjj

  committed
  090f6e5

  Copy full SHA for 090f6e5

Commits on Mar 20, 2022

• track input argument information in one tree at each AOT stage

  Show description for 4b4dc3c
  froystig

  committed
  4b4dc3c

  Copy full SHA for 4b4dc3c

Commits on Mar 18, 2022

• introduce a protocol for compilation wrappers

  froystig

  committed
  e04ea89

  Copy full SHA for e04ea89

Commits on Mar 17, 2022

• mark xmap as an API boundary; filter experimental.maps from tracebacks and source info

  froystig

  committed
  a6841a9

  Copy full SHA for a6841a9

• staging and compilation for custom_transpose

  Show description for 45af307
  

  froystig

  and

  mattjj

  committed
  45af307

  Copy full SHA for 45af307

Commits on Mar 15, 2022

• factor AOT types out to a stages module

  froystig

  committed
  0474884

  Copy full SHA for 0474884

Commits on Mar 10, 2022

• remove _select_and_{gather,scatter}_add from public jax.lax module

  froystig

  committed
  6457279

  Copy full SHA for 6457279

• remove _reduce_window_{min,max,sum,prod} from public jax.lax module

  froystig

  committed
  2f6de4a

  Copy full SHA for 2f6de4a

• remove _convert_element_type from public jax.lax module

  froystig

  committed
  8f93629

  Copy full SHA for 8f93629

Commits on Mar 9, 2022

• remove _dilate_shape from public jax.lax module

  froystig

  committed
  bea7771

  Copy full SHA for bea7771

• remove _ones and _zeros from public jax.lax module

  froystig

  committed
  460feed

  Copy full SHA for 460feed

• remove _delta from public jax.lax module

  froystig

  committed
  0cae316

  Copy full SHA for 0cae316

• remove _tri from public jax.lax module

  froystig

  committed
  90f31c1

  Copy full SHA for 90f31c1

• remove _eye from public jax.lax module

  froystig

  committed
  3c345ee

  Copy full SHA for 3c345ee

• remove _broadcasting_shape_rule from public jax.lax module

  froystig

  committed
  7824325

  Copy full SHA for 7824325

• remove _upcast_fp16_for_computation from public jax.lax module

  froystig

  committed
  2324e5b

  Copy full SHA for 2324e5b

• remove _check_user_dtype_supported from public jax.lax module

  froystig

  committed
  e262c72

  Copy full SHA for e262c72

• remove _reduce_or and _reduce_and from public jax.lax module

  froystig

  committed
  afc8729

  Copy full SHA for afc8729

• remove _reduce_min from public jax.lax module

  froystig

  committed
  c979f64

  Copy full SHA for c979f64

• remove _reduce_max from public jax.lax module

  froystig

  committed
  299d4db

  Copy full SHA for 299d4db

• remove _reduce_sum from public jax.lax module

  froystig

  committed
  6f51957

  Copy full SHA for 6f51957

Commits on Mar 8, 2022

• remove _one and _zero from public jax.lax module

  froystig

  committed
  7890fb7

  Copy full SHA for 7890fb7

• remove three internal functions from public jax.lax module

  Show description for 3f88518
  froystig

  committed
  3f88518

  Copy full SHA for 3f88518

• remove _eq_meet from public jax.lax module

  Show description for 7319982
  

  froystig

  authored and

  jax authors

  committed
  7319982

  Copy full SHA for 7319982

• remove _complex from public jax.lax module

  Show description for 3e77a56
  

  froystig

  authored and

  jax authors

  committed
  3e77a56

  Copy full SHA for 3e77a56

Pagination

Previous

Next