JittorMirror/python/jittor
Dun Liang 7a83dcf09e fix windows mkl bug 2023-04-03 07:56:44 +08:00
..
ccl add ccl 2022-11-06 15:20:54 +08:00
dataset polish dataloader 2023-01-15 13:30:03 +08:00
demo optimize concat and split 2022-04-04 22:53:20 +08:00
einops fix pr#355&add unittest 2022-11-07 12:01:23 +08:00
extern new feature: save_mem 2023-04-02 07:17:55 +08:00
loss3d loss3d: chamfer & emd 2021-06-30 15:22:43 +08:00
math_util add digamma. 2023-03-31 11:05:06 +08:00
models update copyright to 2022 2022-04-22 18:01:02 +08:00
notebook optimize concat and split 2022-04-04 22:53:20 +08:00
other polish log_softmax precision 2022-05-18 15:37:09 +08:00
script update copyright to 2022 2022-04-22 18:01:02 +08:00
src new feature: save_mem 2023-04-02 07:17:55 +08:00
test new feature: save_mem 2023-04-02 07:17:55 +08:00
transform LLM support with jtorch 2023-03-20 22:33:12 +08:00
utils new feature: save_mem 2023-04-02 07:17:55 +08:00
__init__.py fix windows mkl bug 2023-04-03 07:56: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 2022 2022-04-22 18:01:02 +08:00
compile_extern.py fix windows mkl bug 2023-04-03 07:56:44 +08:00
compiler.py support builtin download of onednn because homebrew no longer provides onednn 2.x 2023-01-05 09:38:03 +08:00
contrib.py LLM support with jtorch 2023-03-20 22:33:12 +08:00
depthwise_conv.py 1. fix compilation failure of fp16_compute.h under rocm platform 2022-12-17 21:36:54 +08:00
distributions.py update copyright to 2022 2022-04-22 18:01:02 +08:00
init.py LLM support with jtorch 2023-03-20 22:33:12 +08:00
init_cupy.py update copyright to 2022 2022-04-22 18:01:02 +08:00
linalg.py LLM support with jtorch 2023-03-20 22:33:12 +08:00
lr_scheduler.py update copyright to 2022 2022-04-22 18:01:02 +08:00
misc.py new feature: save_mem 2023-04-02 07:17:55 +08:00
nn.py Merge branch 'master' of github.com:Jittor/jittor 2023-03-31 21:14:20 +08:00
optim.py polish optimzers save&load 2023-01-15 13:38:01 +08:00
pool.py polish backend 2022-10-26 14:29:53 +08:00
pyjt_compiler.py add attr_dict for jittor var 2023-03-24 01:53:30 +08:00
sparse.py update copyright to 2022 2022-04-22 18:01:02 +08:00
version avoid import error when no torch installed 2021-06-18 15:56:13 +08:00