[NVIDIA] Add new SDPA API to jax.nn #56066

Sign in to view logs

Re-run triggered July 5, 2024 08:41

superbobry

#21371

kaixih:jax_sdpa_dev

Status Failure

Total duration 9m 37s

Billable time 16m

Artifacts –

ci-build.yaml

on: pull_request

lint_and_typecheck

Matrix: build

Matrix: Documentation - test code snippets

Matrix: Documentation - render documentation

Matrix: jax2tf_test

Annotations

2 errors

lint_and_typecheck

Process completed with exit code 1.

jax2tf_test (py 3.10 on ubuntu-latest, x64=0)

Process completed with exit code 1.