mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Internal visibility change

Browse Source 
PiperOrigin-RevId: 484340424
This commit is contained in:
Hyeontaek Lim 2022-10-27 13:48:40 -07:00 committed by jax authors
parent 0c119cbb59
commit fc8f40ce0e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
benchmarks/api_benchmark.py
View File

@ -34,6 +34,10 @@ from jax.experimental import pjit as pjit_lib
import jax.numpy as jnp
import numpy as np
from jax.config import config
config.parse_flags_with_absl()
partial = functools.partial