JAX是Autograd和XLA的结合，用于高性能机器学习研究。-Python开发资源-CSDN下载

共475个文件

py：277个

md：56个

ipynb：33个

需积分: 50 13 浏览量 2021-05-24 17:41:57 上传评论收藏 9.62MB ZIP 举报

资源推荐

资源详情

资源评论

收起资源包目录

package

JAX是Autograd和XLA的结合，用于高性能机器学习研究。-Python开发（475个子文件）

BUILD.bazel 2KB

BUILD.bazel 255B

.bazelversion 6B

BUILD 9KB

BUILD 2KB

BUILD 774B

build_defs.bzl 6KB

jax.bzl 2KB

workspace.bzl 1KB

cpu_feature_guard.c 5KB

cusolver.cc 45KB

cusparse.cc 34KB

rocblas.cc 32KB

cublas.cc 11KB

cuda_prng_kernels.cu.cc 4KB

pocketfft.cc 4KB

main.cc 4KB

cuda_lu_pivot_kernels.cu.cc 3KB

cuda_gpu_kernel_helpers.cc 2KB

rocm_gpu_kernel_helpers.cc 2KB

cuda_lu_pivot_kernels.cc 1KB

cuda_prng_kernels.cc 1KB

setup.cfg 314B

setup.cfg 38B

style.css 77B

Dockerfile 2KB

pocketfft.fbs 1009B

wave_movie.gif 5.64MB

.gitignore 219B

handle_pool.h 3KB

kernel_pybind11_helpers.h 2KB

kernel_helpers.h 2KB

cuda_gpu_kernel_helpers.h 1KB

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 142KB

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 50KB

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 815B

LICENSE 11KB

LICENSE 2KB

LICENSE 2KB

LICENSE 1KB

LICENSE_SHORT 575B

xmap_tutorial.md 414KB

autodidax.md 101KB

autodiff_cookbook.md 37KB

Custom_derivative_rules_for_Python_code.md 35KB

README.md 35KB

CHANGELOG.md 32KB

Common_Gotchas_in_JAX.md 31KB

How_JAX_primitives_work.md 26KB

README.md 21KB

custom_derivatives.md 19KB

XLA_in_Python.md 18KB

score_matching.md 16KB

primitives_with_limited_support.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

04-advanced-autodiff.md 12KB

02-jitting.md 12KB

maml.md 12KB

Writing_custom_interpreters_in_Jax.md 12KB

共 475 条

内容反馈

卡卡乐乐

粉丝: 44

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip