JAX:Python+NumPy 程序的可组合转换-开源
需积分: 9 124 浏览量
2021-08-07
09:23:40
上传
评论
收藏 9.75MB ZIP 举报
JAX:Python+NumPy 程序的可组合转换-开源 （490个子文件） 
BUILD.bazel 2KB BUILD.bazel 255B .bazelversion 6B BUILD 10KB BUILD 1KB BUILD 774B build_defs.bzl 6KB jax.bzl 2KB workspace.bzl 1KB cpu_feature_guard.c 5KB cusolver.cc 44KB cusparse.cc 36KB rocblas.cc 32KB cublas.cc 10KB cuda_gpu_kernel_helpers.cc 5KB cuda_prng_kernels.cu.cc 4KB pocketfft.cc 4KB main.cc 4KB cuda_lu_pivot_kernels.cu.cc 3KB rocm_gpu_kernel_helpers.cc 2KB cuda_lu_pivot_kernels.cc 1KB cuda_prng_kernels.cc 1KB setup.cfg 329B setup.cfg 38B style.css 77B Dockerfile 2KB pocketfft.fbs 1009B
wave_movie.gif 5.64MB .gitignore 219B handle_pool.h 3KB cuda_gpu_kernel_helpers.h 2KB kernel_pybind11_helpers.h 2KB kernel_helpers.h 2KB rocm_gpu_kernel_helpers.h 1KB cuda_lu_pivot_kernels.h 1KB cuda_prng_kernels.h 1KB
layout.html 85B pytest.ini 715B mypy.ini 475B score_matching.ipynb 2.43MB convolutions.ipynb 514KB xmap_tutorial.ipynb 422KB maml.ipynb 204KB autodidax.ipynb 144KB Custom_derivative_rules_for_Python_code.ipynb 113KB thinking_in_jax.ipynb 79KB How_JAX_primitives_work.ipynb 59KB autodiff_cookbook.ipynb 56KB Common_Gotchas_in_JAX.ipynb 56KB XLA_in_Python.ipynb 45KB 06-parallelism.ipynb 44KB 01-jax-basics.ipynb 43KB vmapped_log_probs.ipynb 41KB 05.1-pytrees.ipynb 34KB 07-state.ipynb 28KB 02-jitting.ipynb 23KB 04-advanced-autodiff.ipynb 20KB JAX_demo.ipynb 19KB Neural_Network_and_Data_Loading.ipynb 19KB Pmap_Cookbook.ipynb 18KB Writing_custom_interpreters_in_Jax.ipynb 17KB Wave_Equation.ipynb 16KB 05-random-numbers.ipynb 15KB Lorentz_ODE_Solver.ipynb 15KB neural_network_with_tfds_data.ipynb 14KB quickstart.ipynb 14KB JAX_NeurIPS_2020_demo.ipynb 11KB 03-vectorization.ipynb 9KB colab_cpu.ipynb 6KB colab_gpu.ipynb 6KB colab_tpu.ipynb 5KB JAX2TF_getting_started.ipynb 814B LICENSE 11KB LICENSE 2KB LICENSE 2KB LICENSE 1KB xmap_tutorial.md 414KB autodidax.md 103KB README.md 47KB autodiff_cookbook.md 37KB CHANGELOG.md 37KB Custom_derivative_rules_for_Python_code.md 35KB Common_Gotchas_in_JAX.md 32KB How_JAX_primitives_work.md 26KB README.md 22KB custom_derivatives.md 19KB XLA_in_Python.md 18KB score_matching.md 16KB 06-parallelism.md 16KB thinking_in_jax.md 15KB convolutions.md 15KB profiling.md 15KB 01-jax-basics.md 14KB omnistaging.md 14KB primitives_with_limited_support.md 13KB 04-advanced-autodiff.md 12KB 02-jitting.md 12KB maml.md 12KB Writing_custom_interpreters_in_Jax.md 12KB prng.md 11KB共 490 条
- 1
- 2
- 3
- 4
- 5
资源评论
