lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
1,586 Commits 57 Branches 331 Tags 26 MiB
Commit Graph

197 Commits

Author SHA1 Message Date
lzhengning 1119f3cc51 update readme && nn.linear 2023-03-31 19:45:07 +08:00
lzhengning 8a9f0bb904 update load_pytorch.py && add linear 2023-03-31 19:45:07 +08:00
Dun Liang 1e00883ed1 add attr_dict for jittor var 2023-03-24 01:53:30 +08:00
Dun Liang 07f8bd4aff Merge branch 'master' of github.com:Jittor/jittor 2023-03-22 19:47:26 +08:00
Dun Liang db8679748c fp32_guard && ternary broadcast 2023-03-22 19:41:36 +08:00
lzhengning 576b0c9e03 feat: jt.outer & jt.nn.silu 2023-03-21 15:56:21 +08:00
Dun Liang 90850d36ad LLM support with jtorch 2023-03-20 22:33:12 +08:00
Brian Pugh 5dba84ee67 Allow for nn.Sequential to be sliced. 2023-01-15 13:36:29 +08:00
Dun Liang 6fbffb7c62 jtorch compatible 2023-01-06 20:48:04 +08:00
lzhengning 919e863e55 fix depthwise conv error on ROCm 2023-01-04 14:09:53 +08:00
lzhengning d50fe5e754 1. fix compilation failure of fp16_compute.h under rocm platform 
2. add jt.compiler.is_cuda to identify if the current device is CUDA
 2022-12-17 21:36:54 +08:00
Dun Liang 5bc160b19c dataset and dataloader interface polish 2022-11-16 11:31:55 +08:00
Dun Liang 1f06bbf22e fix dropout2d 2022-11-13 16:36:45 +08:00
Dun Liang 74932f3c32 add dropout 2d 2022-11-13 16:17:48 +08:00
lzhengning b2fb32aa52 feats: defaults to keep the first dim in nn.Flatten 2022-09-29 22:14:29 +08:00
Exusial b258cf3a84 add documentation and tests. 2022-09-18 23:41:57 +08:00
Exusial 56255578f9 Add interpolate area support. 2022-09-18 23:41:57 +08:00
uyzhang e0bd748ff1 fix droppath 2022-09-15 16:08:02 +08:00
uyzhang 658ab32bda add droppath in jt.nn 2022-09-14 16:35:38 +08:00
lzhengning 9d176643a0 feat: add KLDivLoss in jittor.nn 2022-09-13 21:22:51 +08:00
lzhengning 3cce690218 feat: add one_hot in jt.nn 2022-09-07 21:48:53 +08:00
Dun Liang 51037cbe9f polish keepdim and keepdims 2022-08-13 14:15:33 +08:00
lzhengning 75b77e411f feature: official ROCm support 2022-08-06 00:44:33 +08:00
Dun Liang 8b1d620b22 Merge branch 'master' of github.com:Jittor/jittor 2022-05-26 14:58:33 +08:00
Dun Liang 31fbabfb6f add th_mode 2022-05-25 17:13:43 +08:00
Dun Liang ba266fa99c polish log_softmax precision 2022-05-18 15:37:09 +08:00
li-xl 9b7a184808 polish cross_entropy_loss 2022-05-13 20:04:50 +08:00
lzhengning cac2b83a1c update copyright to 2022 2022-04-22 18:01:02 +08:00
Dun Liang 0666456a2f v 1.3.3 memory optimization 2022-04-22 15:04:06 +08:00
cxjyxx_me 45df1a4eee fft polish 2022-03-30 01:29:25 -04:00
cxjyxx_me b642b8f1d1 Merge branch 'master' into fft 2022-03-26 23:30:35 -04:00
cxjyxx_me b04ad0ccb4 Merge branch 'master' into fft 2022-03-26 23:22:55 -04:00
Dun Liang 438c8ba692 Merge branch 'master' of github.com:Jittor/jittor 2022-03-15 21:18:08 +08:00
Dun Liang 39ecdd84fd add fp16 support 2022-03-15 17:45:39 +08:00
Dun Liang 81b847e6f0 optimize softmax 2022-02-18 16:45:08 +08:00
yang guo ye e9aca0444c
Update nn.py 
fix typo
 2022-01-30 11:59:38 +08:00
cxjyxx_me e9f681de53 fft 2022-01-25 19:32:06 +08:00
Dun Liang 85ba4898cf add function version conv transpose 2022-01-04 17:29:58 +08:00
Dun Liang ac66897047 polish conv transpose group 2021-12-17 17:54:41 +08:00
Dun Liang 6bc17cb99c polish interpolate 2021-12-06 12:21:18 +08:00
lzhengning 205bd011e7 polish embedding 2021-10-28 13:44:08 +08:00
lzhengning ca2b4f20d0 polish doc for conv, rnn, and activation 2021-10-11 11:54:02 +08:00
lzhengning 9ec300f2aa Polish RNN, GRU, and LSTM 
1. Use Cudnn to speed rnn
2. Fix: document string not correctly rendered
3. Fix: RNN cannot accept relu nonlinearity
4. feat: add default hidden state when executing rnn/gru/lstm
 2021-09-28 17:54:56 +08:00
Dun Liang 4fb462e1b9 polish sub module save load 2021-09-13 11:03:28 +08:00
Dun Liang f807a28e6b add ctcloss 2021-09-08 17:45:17 +08:00
Dun Liang 057dd95658 fix typo 2021-09-08 13:46:51 +08:00
Dun Liang 39165cccfb polish bmm 2021-08-04 19:36:02 +08:00
Dun Liang d482161be0 floor keep type and better mem_info 2021-07-30 20:40:21 +08:00
Dun Liang 0cf77ea11c add bilinear module 2021-07-23 23:31:21 +08:00
cxjyxx_me d89f1419f9 Merge branch 'master' of github.com:Jittor/jittor 2021-07-20 14:16:49 +08:00