.conda_env.yml
.gitignore
.readthedocs.yaml
LICENSE
README.md
changelog.md
makefile
mkdocs.yaml
pyproject.toml
.github/workflows/doc.yaml
.github/workflows/latex.yaml
.github/workflows/lint.yaml
.github/workflows/python-publish.yaml
.github/workflows/test.yaml
docs/api.md
docs/develop.md
docs/gallery_conf.py
docs/gh_requirements.txt
docs/index.md
docs/examples/01_taylor_mode.png
docs/examples/01_taylor_mode.py
docs/examples/02_laplacians.py
docs/examples/03_application_pinns.py
docs/examples/README.md
docs/logo/logo.png
docs/logo/logo.webp
docs/logo/logo_unprocessed.webp
jet/__init__.py
jet/bilaplacian.py
jet/jet_transformer.py
jet/laplacian.py
jet/operations.py
jet/rules.py
jet/signature_parser.py
jet/simplify.py
jet/tracing.py
jet/ttc_coefficients.py
jet/utils.py
jet/vmap.py
jet/weighted_laplacian.py
jet/exp/__init__.py
jet/exp/utils.py
jet/exp/exp01_benchmark_laplacian/__init__.py
jet/exp/exp01_benchmark_laplacian/evaluate_performance.py
jet/exp/exp01_benchmark_laplacian/execute.py
jet/exp/exp01_benchmark_laplacian/plot.py
jet/exp/exp01_benchmark_laplacian/run.py
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_cuda_dim_50_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_cuda_dim_50_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0.pdf
jet/exp/exp01_benchmark_laplacian/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size.pdf
jet/exp/exp01_benchmark_laplacian/gathered/name_bilaplacian_vary_batch_size.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_bilaplacian_vary_num_samples.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_laplacian_vary_batch_size.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_laplacian_vary_num_samples.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_weighted_laplacian_vary_batch_size.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_weighted_laplacian_vary_num_samples.csv
jet/exp/exp01_benchmark_laplacian/gathered/name_weighted_laplacian_vary_rank_ratio.csv
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_laplacian_vary_num_samples/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_weighted_laplacian_vary_num_samples_rank_ratio_1_0/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_name_weighted_laplacian_vary_rank_ratio/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_compiled_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_bilaplacian_vary_num_samples/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_compiled_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_laplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_1/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_0_5/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_weighted_laplacian_vary_batch_size_rank_ratio_1_0/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp01_benchmark_laplacian/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_bilaplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp03_faa_di_bruno/__init__.py
jet/exp/exp03_faa_di_bruno/run.py
jet/exp/exp04_jax_benchmark/__init__.py
jet/exp/exp04_jax_benchmark/evaluate_performance.py
jet/exp/exp04_jax_benchmark/execute.py
jet/exp/exp04_jax_benchmark/plot.py
jet/exp/exp04_jax_benchmark/run.py
jet/exp/exp04_jax_benchmark/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples.pdf
jet/exp/exp04_jax_benchmark/figures/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples.pdf
jet/exp/exp04_jax_benchmark/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size.pdf
jet/exp/exp04_jax_benchmark/figures/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size.pdf
jet/exp/exp04_jax_benchmark/gathered/name_jax_bilaplacian_vary_batch_size.csv
jet/exp/exp04_jax_benchmark/gathered/name_jax_bilaplacian_vary_num_samples.csv
jet/exp/exp04_jax_benchmark/gathered/name_jax_laplacian_vary_batch_size.csv
jet/exp/exp04_jax_benchmark/gathered/name_jax_laplacian_vary_num_samples.csv
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_2048_device_cuda_dim_50_distribution_normal_name_jax_laplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/hessian_trace_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/hessian_trace_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/jet_naive_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/jet_naive_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_batch_size_256_device_cuda_dim_5_distribution_normal_name_jax_bilaplacian_vary_num_samples/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_50_name_jax_laplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/hessian_trace_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/hessian_trace_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/hessian_trace_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_naive_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_naive_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_naive_peakmem_nondifferentiable.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_simplified_best.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_simplified_peakmem.txt
jet/exp/exp04_jax_benchmark/performance/architecture_tanh_mlp_768_768_512_512_1_device_cuda_dim_5_name_jax_bilaplacian_vary_batch_size/jet_simplified_peakmem_nondifferentiable.txt
jet/exp/exp05_example_TTC/jax_TTC.py
jet/exp/exp06_profile_compile/__init__.py
jet/exp/exp06_profile_compile/run.py
jet_for_pytorch.egg-info/PKG-INFO
jet_for_pytorch.egg-info/SOURCES.txt
jet_for_pytorch.egg-info/dependency_links.txt
jet_for_pytorch.egg-info/requires.txt
jet_for_pytorch.egg-info/top_level.txt
paper/icml2024.bst
paper/jet
paper/main.tex
paper/neurips_2025.sty
paper/references.bib
paper/figures/interface_overview.tex
paper/figures/sin_2jet_0.tex
paper/figures/sin_2jet_1.tex
paper/figures/sin_2jet_2.tex
paper/figures/sin_2jet_3.tex
paper/figures/sin_2jet_4.tex
paper/figures/sin_2jet_5.tex
paper/figures/sin_2jet_6.tex
paper/figures/sin_2jet_7.tex
paper/figures/sin_2jet_8.tex
paper/figures/sin_2jet_9.tex
paper/figures/sin_2jet_pull_sum.tex
paper/figures/sin_2jet_push_replicate.tex
paper/figures/taylor_signature.tex
paper/figures/torch_benchmark.tex
paper/figures/torch_benchmark_compiled.tex
paper/figures/ttc_bilaplacian.tex
paper/figures/vanilla_taylor_not_enough.tex
paper/figures/visual_abstract.tex
paper/preamble/custom.tex
paper/preamble/custom_early.tex
paper/preamble/goodfellow.tex
paper/preamble/metadata.tex
paper/preamble/neurips_2025.tex
paper/sections/acknowledgements.tex
paper/sections/appendix_connections.tex
paper/sections/appendix_jax.tex
paper/sections/appendix_linearity_for_taylor.tex
paper/sections/appendix_numerical_analysis.tex
paper/sections/appendix_torch.tex
paper/sections/appendix_ttc.tex
paper/sections/background.tex
paper/sections/checklist.tex
paper/sections/conclusion.tex
paper/sections/experiments.tex
paper/sections/faa_di_bruno.tex
paper/sections/graph_simplifications.tex
paper/sections/introduction.tex
paper/sections/method.tex
paper/tables/torch_benchmark.tex
paper/tables/torch_benchmark_compiled.tex
paper/tables/torch_rank_benchmark.tex
paper/tables/torch_ratios.tex
poster/anim.gif
poster/anim.sh
poster/anim.tex
poster/main.tex
poster/figures/overview.pdf
poster/figures/sin_2jet_0.pdf
poster/figures/sin_2jet_1.pdf
poster/figures/sin_2jet_2.pdf
poster/figures/sin_2jet_3.pdf
poster/figures/taylor.pdf
poster/logos/logo-concordia.pdf
poster/logos/logo-eth-inverted.png
poster/logos/logo-hu-berlin.pdf
poster/logos/logo-mila.png
poster/logos/logo-zib.png
poster/postertheme/colors/vector_institute.tex
poster/postertheme/vector_institute/vector_institute.sty
poster/postertheme/vector_institute/assets/logo_horizontal_inverted.png
poster/tables/benchmark.tex
test/__init__.py
test/test___init__.py
test/test_bilaplacian.py
test/test_laplacian.py
test/test_nesting.py
test/test_rules.py
test/test_signature_parser.py
test/test_simplify.py
test/test_ttc_coefficients.py
test/test_utils.py
test/test_vmap.py
test/utils.py
test/exp/__init__.py
test/exp/test_exp01_benchmark_laplacian.py
test/exp/test_exp03.py