JittorMirror/python/jittor
DongYang Li 7416cfb7e4
update version
2024-06-25 16:49:44 +08:00
..
ccl add ccl 2022-11-06 15:20:54 +08:00
compatibility polish rocm support 2024-05-20 21:34:26 +08:00
dataset Merge pull request #443 from co63oc/patch-1 2024-05-16 15:20:07 +08:00
demo optimize concat and split 2022-04-04 22:53:20 +08:00
einops madd bf16 2023-09-08 22:43:31 +08:00
extern Merge branch 'master' into master 2024-05-21 12:48:58 +08:00
loss3d loss3d: chamfer & emd 2021-06-30 15:22:43 +08:00
math_util add Gamma Distribution on Cuda (#435) 2023-04-25 14:25:10 +08:00
models update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
notebook optimize concat and split 2022-04-04 22:53:20 +08:00
other madd bf16 2023-09-08 22:43:31 +08:00
script MPI half type 2023-08-02 00:14:39 +08:00
src Merge branch 'master' into master 2024-05-21 12:48:58 +08:00
test add complex matmul, inv, qr, eig, and svd 2024-05-08 21:09:07 +08:00
transform update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
utils polish computing graph liveness 2023-09-25 14:48:05 +08:00
vcompiler add fallback_func for static graph compiler 2024-01-03 03:57:14 +08:00
__init__.py update version 2024-06-25 16:49:44 +08:00
__init__.pyi 1. polish index_op doc; 2. polish pyi files 2022-09-14 16:09:14 +08:00
attention.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
compile_extern.py MPI half type 2023-08-02 00:14:39 +08:00
compiler.py add acl support [version 1.3.8.7] 2023-07-13 14:36:57 +08:00
contrib.py fix illegal parameters of PixelShuffle of issue #458,fix validity of concat of issue #459 2024-05-30 19:03:17 +08:00
depthwise_conv.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
distributions.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
init.py support scalar in getitem && fix bugs for jtorch 2023-06-19 16:06:19 +08:00
init_cupy.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
linalg.py add complex matmul, inv, qr, eig, and svd 2024-05-08 21:09:07 +08:00
lr_scheduler.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
misc.py fix: jt.Var.expand with valid index -1 2024-05-29 11:05:53 +08:00
nn.py polish nn.Sequential attribute 2024-06-05 22:31:20 +08:00
optim.py dont deep copy optim state dict 2023-09-15 13:49:13 +08:00
pool.py check x.shape and kernel_size of Pool and Pool3d,issue #461,#463 2024-05-31 14:33:07 +08:00
pyjt_compiler.py fitten code 2023-12-05 01:34:26 +08:00
sparse.py update copyright to 2023; update version for PR https://github.com/Jittor/jittor/pull/435 2023-04-25 14:28:38 +08:00
version avoid import error when no torch installed 2021-06-18 15:56:13 +08:00
weightnorm.py add support for weight_norm and histc. (#420) 2023-04-05 21:59:20 +08:00