truesoni/jax

main

jax

Grade	Name	Complexity	Churn	Issues
F	jax\_src\numpy\lax_numpy.py	1248	17	17
D	jax\_src\lax\lax.py	1073	6	15
F	tests\api_test.py	914	12	101
F	jax\_src\core.py	852	5	33
F	jaxlib\mosaic\python\apply_vector_layout.py	749	12	31
F	jax\_src\interpreters\partial_eval.py	720	4	40
A	tests\lax_numpy_test.py	633	11	3
D	jax\_src\interpreters\pxla.py	604	13	14
C	jax\experimental\sparse\bcoo.py	532	3	15
B	docs\autodidax.py	482	2	18
D	jax\_src\interpreters\mlir.py	480	14	14
A	tests\pjit_test.py	461	10	-
A	jax\experimental\shard_map.py	450	3	1
C	jax\experimental\jax2tf\jax2tf.py	397	9	16
B	jax\_src\pjit.py	387	8	9
A	tests\pmap_test.py	377	5	3
B	jaxlib\mosaic\dialect\tpu\transforms\infer_vector_layout.cc	362	7	8
C	jax\_src\lax\control_flow\loops.py	353	5	17
B	jax\experimental\export\shape_poly.py	352	7	10
B	tests\lax_test.py	342	3	12