Search results

[Mosaic] apply_vector_layout C++ rewrite (17): for.op (from cl/568376871)

Status: Closed (abandoned).

jax-ml/jaxnumber 17872

#17872 In jax-ml/jax;

· by copybara-service was closed on Oct 5, 2023

[Mosaic] apply_vector_layout C++ rewrite: Have each rule use its own builder instead of passing a builder in the context

Status: Closed (abandoned).

jax-ml/jaxnumber 17871

#17871 In jax-ml/jax;

· by copybara-service was closed on Oct 5, 2023

[Mosaic] infer_memref_layout C++ rewrite

Status: Closed (abandoned).

jax-ml/jaxnumber 17940

#17940 In jax-ml/jax;

· by copybara-service was closed on Oct 5, 2023

[Mosaic] apply_vector_layout C++ rewrite: various bug fixes

Status: Closed (abandoned).

jax-ml/jaxnumber 17939

#17939 In jax-ml/jax;

· by copybara-service was closed on Oct 5, 2023

[Mosaic] apply_vector_layout: Use shape in generalizes check (in Python)

Status: Closed (abandoned).

jax-ml/jaxnumber 17942

#17942 In jax-ml/jax;

· by copybara-service was closed on Oct 5, 2023

[Mosaic] apply_vector_layout C++ rewrite: Handle elementwise ops by checking for the Elementwise trait and using the generic Operation interface, without templates

Status: Closed (abandoned).

jax-ml/jaxnumber 17937

#17937 In jax-ml/jax;

· by copybara-service was closed on Oct 9, 2023

[Mosaic] apply_vector_layout C++ rewrite (18): vector.transpose

Status: Closed (abandoned).

jax-ml/jaxnumber 17914

#17914 In jax-ml/jax;

· by copybara-service was closed on Oct 9, 2023

[JAX] randint goes from [min, max) so [min - N, max + M] actually maps to [min,

Status: Merged (completed).

jax-ml/jaxnumber 18021

#18021 In jax-ml/jax;

· by copybara-service was closed on Oct 9, 2023

[Mosaic] apply_vector_layout C++ rewrite: Fix check for defining layout in disassemble

Status: Closed (abandoned).

jax-ml/jaxnumber 17943

#17943 In jax-ml/jax;

· by copybara-service was closed on Oct 10, 2023

Update README.md

cla: yes

Status: Merged (completed).

jax-ml/jaxnumber 50

#50 In jax-ml/jax;

· by alexbw was closed on Dec 11, 2018

add batching rules, fix jacfwd/jacrev transpose bug

cla: yes

Status: Merged (completed).

jax-ml/jaxnumber 82

#82 In jax-ml/jax;

· by mattjj was closed on Dec 12, 2018

Disable tests that don't pass on all backends.

cla: yes

Status: Merged (completed).

jax-ml/jaxnumber 109

#109 In jax-ml/jax;

· by hawkinsp was closed on Dec 14, 2018