Commits

Branch selector

 main

User selector

chr1sj0nes

Datepicker

All time

Commit History

Commits on Apr 13, 2021

• Simplify all-reduce translation rule.

  Show description for f87d05b
  chr1sj0nes

  committed
  f87d05b

  Copy full SHA for f87d05b

Commits on Mar 2, 2021

• Lower remat to while(int(rng(1, 2))) on GPU.

  chr1sj0nes

  committed
  9238720

  Copy full SHA for 9238720

Commits on Jan 26, 2021

• [JAX-GPU] Use XLA AllToAll op on GPU (for supported configurations).

  chr1sj0nes

  committed
  b633898

  Copy full SHA for b633898

Commits on Jan 19, 2021

• Use XLA AllGather op for GPU (attempt 2).

  Show description for 4b48c7f
  chr1sj0nes

  committed
  4b48c7f

  Copy full SHA for 4b48c7f

Commits on Jan 12, 2021

• [JAX] Use XLA AllGather op for GPU (when supported).

  chr1sj0nes

  committed
  474fdfc

  Copy full SHA for 474fdfc

Commits on Jan 6, 2021

• Skip tests that are failing without omnistaging (which will not be an option shortly).

  Show description for 3896ac2
  chr1sj0nes

  committed
  3896ac2

  Copy full SHA for 3896ac2

Commits on Jul 9, 2020

• Add simple JAX API microbenchmarks. (jax-ml#3679)

  chr1sj0nes

  authored
  c1aeb8b

  Copy full SHA for c1aeb8b

Commits on Apr 28, 2020

• Remove unused ispure method (jax-ml#2781)

  chr1sj0nes

  authored
  2d96cfb

  Copy full SHA for 2d96cfb

Commits on Apr 16, 2020

• Merge branch 'master' into changelist/306845248

  chr1sj0nes

  authored
  f7070e5

  Copy full SHA for f7070e5

• Some cleanup and reformatting in xla.py.

  Show description for 903b50e
  chr1sj0nes

  committed
  903b50e

  Copy full SHA for 903b50e

• Fix copy-paste error

  Show description for 40884db
  chr1sj0nes

  authored
  40884db

  Copy full SHA for 40884db

Commits on Mar 9, 2020

• Fix a few type annotations in api.py. (jax-ml#2387)

  chr1sj0nes

  authored
  0080c89

  Copy full SHA for 0080c89

• Give Vars an aval. (jax-ml#2299)

  chr1sj0nes

  authored
  1e7d13b

  Copy full SHA for 1e7d13b

Commits on Feb 24, 2020

• Minor dedupe xla/pxla tracing code. (jax-ml#2258)

  chr1sj0nes

  authored
  8028e4b

  Copy full SHA for 8028e4b

Commits on Jan 29, 2020

• Add num_partitions parameter to get_compile_options. (jax-ml#2052)

  chr1sj0nes

  authored
  a503395

  Copy full SHA for a503395

Commits on Nov 27, 2019

• Fix variable name in error message

  chr1sj0nes

  authored
  43a1c00

  Copy full SHA for 43a1c00

Commits on Nov 25, 2019

• Remove join_pvals

  Show description for ace14bb
  

  chr1sj0nes

  authored and

  mattjj

  committed
  ace14bb

  Copy full SHA for ace14bb

Commits on Nov 22, 2019

• Remove as_abstract_val.

  

  chr1sj0nes

  authored and

  mattjj

  committed
  20a7e7b

  Copy full SHA for 20a7e7b

• Fix as_abstract_value.

  Show description for 9c966f9
  

  chr1sj0nes

  authored and

  mattjj

  committed
  9c966f9

  Copy full SHA for 9c966f9