Path	Lines of Code
benchmarks/operator_authoring.py	220
benchmarks/per_sample_grads.py	71
benchmarks/pointwise_scorecard.py	171
benchmarks/process_scorecard.py	16
benchmarks/transformer_fusion_patterns/__init__.py	1
benchmarks/transformer_fusion_patterns/benchmark.py	124
benchmarks/transformer_fusion_patterns/benchmark_helper.py	115
benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py	42
codegen/codegen_outofplacebatching.py	185
codegen/gen_functorch_lagging_op_db.py	22
codegen/gen_plumbing.py	46
functorch/__init__.py	71
functorch/_src/__init__.py	1
functorch/_src/aot_autograd.py	430
functorch/_src/compilers.py	186
functorch/_src/custom_function.py	13
functorch/_src/decompositions.py	187
functorch/_src/eager_transforms.py	386
functorch/_src/fx_minifier.py	187
functorch/_src/make_functional.py	187
functorch/_src/operator_authoring.py	360
functorch/_src/python_key.py	168
functorch/_src/pytree_hacks.py	12
functorch/_src/top_operators_github_usage.py	610
functorch/_src/vmap.py	155
functorch/compile/__init__.py	17
functorch/csrc/BatchRulesBinaryOps.cpp	314
functorch/csrc/BatchRulesConvolution.cpp	347
functorch/csrc/BatchRulesDecompositions.cpp	214
functorch/csrc/BatchRulesDynamic.cpp	53
functorch/csrc/BatchRulesFactory.cpp	58
functorch/csrc/BatchRulesHelper.cpp	97
functorch/csrc/BatchRulesHelper.h	347
functorch/csrc/BatchRulesLinearAlgebra.cpp	149
functorch/csrc/BatchRulesLoss.cpp	249
functorch/csrc/BatchRulesModules.cpp	324
functorch/csrc/BatchRulesNorm.cpp	720
functorch/csrc/BatchRulesPooling.cpp	57
functorch/csrc/BatchRulesReduceOps.cpp	262
functorch/csrc/BatchRulesScatterOps.cpp	462
functorch/csrc/BatchRulesUnaryOps.cpp	111
functorch/csrc/BatchRulesViews.cpp	329
functorch/csrc/BatchedFallback.cpp	276
functorch/csrc/BatchedFallback.h	40
functorch/csrc/BatchedTensorImpl.cpp	124
functorch/csrc/BatchedTensorImpl.h	71
functorch/csrc/BatchingMetaprogramming.h	107
functorch/csrc/BatchingRegistrations.cpp	565
functorch/csrc/CompileCache.cpp	175
functorch/csrc/CompileCache.h	7
functorch/csrc/Constants.h	18
functorch/csrc/CustomFunction.cpp	242
functorch/csrc/CustomFunction.h	7
functorch/csrc/DynamicLayer.cpp	517
functorch/csrc/DynamicLayer.h	42
functorch/csrc/InPlacePlumbing.h	33
functorch/csrc/OutOfPlacePlumbing.h	5
functorch/csrc/PlumbingHelper.cpp	30
functorch/csrc/PlumbingHelper.h	9
functorch/csrc/PointwiseOperatorCompileCache.cpp	584
functorch/csrc/PointwiseOperatorCompileCache.h	7
functorch/csrc/PyTorchOperatorHacks.cpp	157
functorch/csrc/TensorWrapper.cpp	154
functorch/csrc/TensorWrapper.h	44
functorch/csrc/VmapModeRegistrations.cpp	85
functorch/csrc/VmapTransforms.cpp	173
functorch/csrc/VmapTransforms.h	41
functorch/csrc/init.cpp	213
functorch/experimental/__init__.py	1
notebooks/_src/plot_ensembling.py	44
notebooks/_src/plot_jacobians_and_hessians.py	66
notebooks/_src/plot_per_sample_gradients.py	62
op_analysis/gen_data.py	141
setup.cfg	14
setup.py	99