truesoni/jax

main

jax

Grade	Name	Complexity	Churn	Issues
A	jax\_src\pallas\mosaic\lowering.py	331	8	6
B	jax\_src\lax\linalg.py	305	4	10
A	jaxlib\mosaic\dialect\tpu\transforms\apply_vector_layout.cc	305	23	1
A	jax\_src\lax\slicing.py	304	4	6
B	jax\_src\lax\parallel.py	303	6	10
B	jax\_src\maps.py	297	5	7
A	jax\_src\sharding_impls.py	296	3	1
A	tests\xmap_test.py	267	5	1
A	jax\_src\test_util.py	250	9	2
A	jax\_src\pallas\triton\lowering.py	249	19	3
C	jax\_src\prng.py	243	10	16
B	tests\lax_control_flow_test.py	242	2	11
B	jax\_src\random.py	236	10	10
B	jax\_src\api.py	235	15	8
A	tests\host_callback_test.py	219	2	4
B	jax\experimental\jax2tf\tests\shape_poly_test.py	209	8	10
A	jax\_src\interpreters\ad.py	208	3	6
A	tests\random_lax_test.py	207	6	9
A	jax\_src\lax\control_flow\conditionals.py	202	4	12
D	tests\linalg_test.py	193	7	25