xla
[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper
#9180

Merged

[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper #9180

yaochengji merged 2 commits into master from chengji/gmm-use-call_jax

bhavya01 requested a review from

miladm 1 year ago

yaochengji requested a review from

tengyifei 1 year ago

yaochengji requested a review from

qihqi 1 year ago

yaochengji requested a review from

lsy323 1 year ago

yaochengji requested a review from

vanbasten23 1 year ago

vanbasten23 commented on 2025-05-16

[Kernel] Use call_jax to simplify the gmm pallas kernel wrapper

27a4c0a5

yaochengji force pushed from b7965145 to 27a4c0a5 1 year ago

Merge branch 'master' into chengji/gmm-use-call_jax

2d07b534

vanbasten23 commented on 2025-05-16

vanbasten23 approved these changes on 2025-05-16

yaochengji merged f39434ab into master 1 year ago

Reviewers

vanbasten23

miladm

tengyifei

qihqi

lsy323

Assignees

No one assigned

Labels

None yet

Milestone

No milestone