Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 15, 2020

• Merge pull request jax-ml#5191 from skye:fix_colab_tpu

  Show description for 34bc6ca
  

  jax authors

  committed
  34bc6ca

  Copy full SHA for 34bc6ca

• Fix colab_tpu.setup_tpu import in example Cloud TPU notebooks.

  skye

  committed
  85796cc

  Copy full SHA for 85796cc

Commits on Dec 14, 2020

• Merge pull request jax-ml#5189 from hawkinsp:herbie

  Show description for 0ca612c
  

  jax authors

  committed
  0ca612c

  Copy full SHA for 0ca612c

• Merge pull request jax-ml#5183 from AntoinePlumerault:patch-1

  Show description for 4390f3b
  

  jax authors

  committed
  4390f3b

  Copy full SHA for 4390f3b

• Merge pull request jax-ml#5177 from minoring:pad-reflect-type

  Show description for 931f925
  

  jax authors

  committed
  931f925

  Copy full SHA for 931f925

• Improve a few JVP rules with rewrites from Herbie.

  hawkinsp

  committed
  de8df3a

  Copy full SHA for de8df3a

• Merge pull request jax-ml#5163 from bchetioui:factor_test_utils

  Show description for a28436b
  

  jax authors

  committed
  a28436b

  Copy full SHA for a28436b

• Merge pull request jax-ml#5137 from hawkinsp:dot

  Show description for 7ac3c04
  

  jax authors

  committed
  7ac3c04

  Copy full SHA for 7ac3c04

Commits on Dec 13, 2020

• Typo ?

  Show description for 269676e
  AntoinePlumerault

  authored
  269676e

  Copy full SHA for 269676e

• Set collect_limitations variable to True and remove original_impl.

  bchetioui

  committed
  3a06dfe

  Copy full SHA for 3a06dfe

• Remove bool from odd reflect_type test

  minoring

  committed
  c1d8673

  Copy full SHA for c1d8673

Commits on Dec 12, 2020

• Merge pull request jax-ml#5121 from qiuminxu:add_tflite_example

  Show description for 91a0dd7
  

  jax authors

  committed
  91a0dd7

  Copy full SHA for 91a0dd7

• Merge pull request jax-ml#5161 from qiuminxu:jax2tf_avg_pool

  Show description for b85e059
  

  jax authors

  committed
  b85e059

  Copy full SHA for b85e059

• Add reflect_type argument for symmetric and reflect modes in padding

  minoring

  committed
  f0a248c

  Copy full SHA for f0a248c

• Merge pull request jax-ml#5160 from jakevdp:weak-types-slicing

  Show description for 4d3e0fd
  

  jax authors

  committed
  4d3e0fd

  Copy full SHA for 4d3e0fd

Commits on Dec 11, 2020

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

  Show description for e802bf9
  skye

  committed
  e802bf9

  Copy full SHA for e802bf9

• Preserve weak types in array slices

  jakevdp

  committed
  ebaa35e

  Copy full SHA for ebaa35e

• Merge pull request jax-ml#5172 from jakevdp:boilerplate

  Show description for a5df588
  

  jax authors

  committed
  a5df588

  Copy full SHA for a5df588

• Cleanup: remove unnecessary code in convert_element_type

  jakevdp

  committed
  de96856

  Copy full SHA for de96856

• Cleanup: remove remaining instances of rng_factory boilerplate

  jakevdp

  committed
  1a83bb6

  Copy full SHA for 1a83bb6

• Merge pull request jax-ml#5138 from jakevdp:boilerplate

  Show description for 92ae9d8
  

  jax authors

  committed
  92ae9d8

  Copy full SHA for 92ae9d8

• Merge pull request jax-ml#5171 from apaszke:relax-check

  Show description for 940b197
  

  jax authors

  committed
  940b197

  Copy full SHA for 940b197

• Update mnist.py

  qiuminxu

  committed
  542f33e

  Copy full SHA for 542f33e

• Check for non-tuples and not int subclasses in Chunked

  Show description for 146c6eb
  apaszke

  committed
  146c6eb

  Copy full SHA for 146c6eb

• Added a non-XLA conversion path for reduce_window_sum

  qiuminxu

  committed
  aee42a3

  Copy full SHA for aee42a3

• Merge pull request jax-ml#5099 from apaszke:xmap-multiple-mesh-dims

  Show description for ca46894
  

  jax authors

  committed
  ca46894

  Copy full SHA for ca46894

• Merge pull request jax-ml#5157 from jakevdp:weak-types-creation

  Show description for 7ee2710
  

  jax authors

  committed
  7ee2710

  Copy full SHA for 7ee2710

• Merge pull request jax-ml#5168 from hawkinsp:jaxlib

  Show description for ba62270
  

  jax authors

  committed
  ba62270

  Copy full SHA for ba62270

• Export LICENSE.txt file in jaxlib wheels.

  hawkinsp

  committed
  21914db

  Copy full SHA for 21914db

• Include a LICENSE.txt in jaxlib wheels.

  Show description for c23edb8
  

  hawkinsp

  authored and

  jax authors

  committed
  c23edb8

  Copy full SHA for c23edb8

• Support mapping a single logical axis to multiple mesh axes in xmap

  apaszke

  committed
  8fcacd6

  Copy full SHA for 8fcacd6

• [jax2tf] Factorize and clean up test utils.

  bchetioui

  committed
  b6114a0

  Copy full SHA for b6114a0

• Merge pull request jax-ml#5146 from apaszke:xmap-api-update

  Show description for 6841d8f
  

  jax authors

  committed
  6841d8f

  Copy full SHA for 6841d8f

• Make vectorization the default xmap schedule

  Show description for 8438ffe
  apaszke

  committed
  8438ffe

  Copy full SHA for 8438ffe

• Make xmap in/out_axes take in native Python containers

  Show description for 136f9ec
  apaszke

  committed
  136f9ec

  Copy full SHA for 136f9ec

Pagination

Previous

Next