Search results

[mosaic_gpu] LayoutTest now correctly resets the value of MOSAIC_GPU_DUMP_SASS

Status: Open (in progress).

jax-ml/jaxnumber 26273

#26273 In jax-ml/jax;

· copybara-service opened on Feb 3, 2025

[pallas:triton] Fallback lowering rules for math functions now use general dtypes

Status: Open (in progress).

jax-ml/jaxnumber 26271

#26271 In jax-ml/jax;

· copybara-service opened on Feb 3, 2025

Jax 0.5.0 error: no member named 'Auto' in 'llvm::ms_demangle::PrimitiveKind'

bugSomething isn't workingSomething isn't working

Status: Open.

#26143 In jax-ml/jax;

· gongomgra opened on Jan 28, 2025

[pallas:triton] The lowering now uses PTX instead of Triton IR

Status: Open (in progress).

jax-ml/jaxnumber 26081

#26081 In jax-ml/jax;

· copybara-service opened on Jan 24, 2025

Clarify type hints for wrapped functions

pull readyReady for copybara import and testingReady for copybara import and testing

Status: Open (in progress).

jax-ml/jaxnumber 25994

#25994 In jax-ml/jax;

· ncoghlan opened on Jan 21, 2025

[pallas:triton] Removed unused serialized_metadata field from TritonCompilerParams

Status: Open (in progress).

jax-ml/jaxnumber 25552

#25552 In jax-ml/jax;

· copybara-service opened on Dec 17, 2024

Allow Pallas Triton to be serialized as PTX

enhancementNew feature or requestNew feature or request

Status: Open.

#25196 In jax-ml/jax;

· sbodenstein opened on Dec 2, 2024

[pallas:mosaic_gpu] Cache IR constants in the lowering

Status: Open (in progress).

jax-ml/jaxnumber 24944

#24944 In jax-ml/jax;

· copybara-service opened on Nov 18, 2024

Fix Pyright issue with the type of NotMapped

Status: Open (in progress).

jax-ml/jaxnumber 23492

#23492 In jax-ml/jax;

· knyazer opened on Sep 7, 2024

Unsupported conversion from f64 to f16 in pallas despite not using fp16

bugSomething isn't workingSomething isn't working

pallasIssues pertaining to Pallas (GPU or TPU)Issues pertaining to Pallas (GPU or TPU)

Status: Open.

#23179 In jax-ml/jax;

· lengstrom opened on Aug 21, 2024

Incorrect Jacobian of jax.scipy.special.logsumexp when b contains a 0

bugSomething isn't workingSomething isn't working

Status: Open.

#22398 In jax-ml/jax;

· Adrien-Kahn opened on Jul 11, 2024

Support dlpack pinned host memory device type (kDLCPUPinned / kDLCUDAHost)

bugSomething isn't workingSomething isn't working

Status: Open.

#22346 In jax-ml/jax;

· JesseFarebro opened on Jul 9, 2024