JittorMirror/python/jittor
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
..
dataset Merge branch 'master' of github.com:Jittor/jittor 2021-09-15 17:45:37 +08:00
extern add fp16 support 2022-03-15 17:45:39 +08:00
loss3d loss3d: chamfer & emd 2021-06-30 15:22:43 +08:00
models add fp16 support 2022-03-15 17:45:39 +08:00
notebook add learn jittor in 60 min 2021-07-14 21:51:52 +08:00
other add fp16 support 2022-03-15 17:45:39 +08:00
script support aarch64 uos 2021-06-20 15:03:28 +08:00
src add fp16 support 2022-03-15 17:45:39 +08:00
test add fp16 support 2022-03-15 17:45:39 +08:00
transform fix: to_tensor applied on a jt.Var 2021-11-09 23:00:57 +08:00
utils add fp16 support 2022-03-15 17:45:39 +08:00
__init__.py add fp16 support 2022-03-15 17:45:39 +08:00
__init__.pyi polish gen_pyi 2021-10-26 20:14:17 +08:00
attention.py update 2020 -> 2021 2021-01-07 16:25:00 +08:00
compile_extern.py add fp16 support 2022-03-15 17:45:39 +08:00
compiler.py fp16 support, opbytype interface 2022-02-20 20:34:49 +08:00
contrib.py add bool setitem alias and vector_to_tuple 2022-01-10 12:36:28 +08:00
depthwise_conv.py add stride padding(h,w) support and ctrl-c quick exit 2021-02-20 16:34:24 +08:00
distributions.py floor keep type and better mem_info 2021-07-30 20:40:21 +08:00
init.py Add init.trunc_normal_ 2022-01-04 16:09:57 +08:00
init_cupy.py failback to mpi if nccl not available 2021-05-14 13:45:38 +08:00
linalg.py msvc support 2021-09-15 17:34:50 +08:00
lr_scheduler.py add exp lr and remove warning as error 2021-07-26 21:21:20 +08:00
misc.py add fp16 support 2022-03-15 17:45:39 +08:00
nn.py add fp16 support 2022-03-15 17:45:39 +08:00
optim.py add dependency for mpi ops 2021-10-15 16:22:56 +08:00
pool.py add fp16 support 2022-03-15 17:45:39 +08:00
pyjt_compiler.py polish error msg 2021-10-18 15:37:02 +08:00
sparse.py update 2020 -> 2021 2021-01-07 16:25:00 +08:00
version avoid import error when no torch installed 2021-06-18 15:56:13 +08:00