@elaraai/east-py-datascience 0.0.2-beta.8 → 0.0.2-beta.80

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (123) hide show
  1. package/README.md +58 -1
  2. package/dist/src/alns/alns.d.ts +528 -0
  3. package/dist/src/alns/alns.d.ts.map +1 -0
  4. package/dist/src/alns/alns.js +238 -0
  5. package/dist/src/alns/alns.js.map +1 -0
  6. package/dist/src/google_or/google_or.d.ts +2422 -0
  7. package/dist/src/google_or/google_or.d.ts.map +1 -0
  8. package/dist/src/google_or/google_or.js +542 -0
  9. package/dist/src/google_or/google_or.js.map +1 -0
  10. package/dist/{gp → src/gp}/gp.d.ts +185 -136
  11. package/dist/src/gp/gp.d.ts.map +1 -0
  12. package/dist/{gp → src/gp}/gp.js +64 -12
  13. package/dist/src/gp/gp.js.map +1 -0
  14. package/dist/src/index.d.ts +34 -0
  15. package/dist/src/index.d.ts.map +1 -0
  16. package/dist/src/index.js +57 -0
  17. package/dist/src/index.js.map +1 -0
  18. package/dist/src/lightgbm/lightgbm.d.ts +575 -0
  19. package/dist/src/lightgbm/lightgbm.d.ts.map +1 -0
  20. package/dist/{lightgbm → src/lightgbm}/lightgbm.js +104 -18
  21. package/dist/src/lightgbm/lightgbm.js.map +1 -0
  22. package/dist/src/lightning/lightning.d.ts +1594 -0
  23. package/dist/src/lightning/lightning.d.ts.map +1 -0
  24. package/dist/src/lightning/lightning.js +468 -0
  25. package/dist/src/lightning/lightning.js.map +1 -0
  26. package/dist/{mads → src/mads}/mads.d.ts +109 -112
  27. package/dist/src/mads/mads.d.ts.map +1 -0
  28. package/dist/{mads → src/mads}/mads.js +6 -8
  29. package/dist/src/mads/mads.js.map +1 -0
  30. package/dist/src/mapie/mapie.d.ts +3680 -0
  31. package/dist/src/mapie/mapie.d.ts.map +1 -0
  32. package/dist/src/mapie/mapie.js +616 -0
  33. package/dist/src/mapie/mapie.js.map +1 -0
  34. package/dist/{ngboost → src/ngboost}/ngboost.d.ts +192 -142
  35. package/dist/src/ngboost/ngboost.d.ts.map +1 -0
  36. package/dist/{ngboost → src/ngboost}/ngboost.js +67 -14
  37. package/dist/src/ngboost/ngboost.js.map +1 -0
  38. package/dist/src/optimization/optimization.d.ts +420 -0
  39. package/dist/src/optimization/optimization.d.ts.map +1 -0
  40. package/dist/src/optimization/optimization.js +257 -0
  41. package/dist/src/optimization/optimization.js.map +1 -0
  42. package/dist/{optuna → src/optuna}/optuna.d.ts +374 -314
  43. package/dist/src/optuna/optuna.d.ts.map +1 -0
  44. package/dist/{optuna → src/optuna}/optuna.js +2 -0
  45. package/dist/src/optuna/optuna.js.map +1 -0
  46. package/dist/src/pymc/pymc.d.ts +2932 -0
  47. package/dist/src/pymc/pymc.d.ts.map +1 -0
  48. package/dist/src/pymc/pymc.js +688 -0
  49. package/dist/src/pymc/pymc.js.map +1 -0
  50. package/dist/src/scipy/scipy.d.ts +2205 -0
  51. package/dist/src/scipy/scipy.d.ts.map +1 -0
  52. package/dist/src/scipy/scipy.js +884 -0
  53. package/dist/src/scipy/scipy.js.map +1 -0
  54. package/dist/src/shap/shap.d.ts +2988 -0
  55. package/dist/src/shap/shap.d.ts.map +1 -0
  56. package/dist/src/shap/shap.js +500 -0
  57. package/dist/src/shap/shap.js.map +1 -0
  58. package/dist/{simanneal → src/simanneal}/simanneal.d.ts +257 -160
  59. package/dist/src/simanneal/simanneal.d.ts.map +1 -0
  60. package/dist/{simanneal → src/simanneal}/simanneal.js +105 -8
  61. package/dist/src/simanneal/simanneal.js.map +1 -0
  62. package/dist/src/simulation/simulation.d.ts +431 -0
  63. package/dist/src/simulation/simulation.d.ts.map +1 -0
  64. package/dist/src/simulation/simulation.js +306 -0
  65. package/dist/src/simulation/simulation.js.map +1 -0
  66. package/dist/src/sklearn/sklearn.d.ts +6362 -0
  67. package/dist/src/sklearn/sklearn.d.ts.map +1 -0
  68. package/dist/src/sklearn/sklearn.js +1508 -0
  69. package/dist/src/sklearn/sklearn.js.map +1 -0
  70. package/dist/src/torch/torch.d.ts +1205 -0
  71. package/dist/src/torch/torch.d.ts.map +1 -0
  72. package/dist/{torch → src/torch}/torch.js +109 -18
  73. package/dist/src/torch/torch.js.map +1 -0
  74. package/dist/src/types.d.ts +43 -0
  75. package/dist/src/types.d.ts.map +1 -0
  76. package/dist/src/types.js +44 -0
  77. package/dist/src/types.js.map +1 -0
  78. package/dist/src/xgboost/xgboost.d.ts +1424 -0
  79. package/dist/src/xgboost/xgboost.d.ts.map +1 -0
  80. package/dist/src/xgboost/xgboost.js +432 -0
  81. package/dist/src/xgboost/xgboost.js.map +1 -0
  82. package/package.json +12 -12
  83. package/dist/gp/gp.d.ts.map +0 -1
  84. package/dist/gp/gp.js.map +0 -1
  85. package/dist/index.d.ts +0 -27
  86. package/dist/index.d.ts.map +0 -1
  87. package/dist/index.js +0 -41
  88. package/dist/index.js.map +0 -1
  89. package/dist/lightgbm/lightgbm.d.ts +0 -494
  90. package/dist/lightgbm/lightgbm.d.ts.map +0 -1
  91. package/dist/lightgbm/lightgbm.js.map +0 -1
  92. package/dist/mads/mads.d.ts.map +0 -1
  93. package/dist/mads/mads.js.map +0 -1
  94. package/dist/ngboost/ngboost.d.ts.map +0 -1
  95. package/dist/ngboost/ngboost.js.map +0 -1
  96. package/dist/optuna/optuna.d.ts.map +0 -1
  97. package/dist/optuna/optuna.js.map +0 -1
  98. package/dist/scipy/scipy.d.ts +0 -1260
  99. package/dist/scipy/scipy.d.ts.map +0 -1
  100. package/dist/scipy/scipy.js +0 -413
  101. package/dist/scipy/scipy.js.map +0 -1
  102. package/dist/shap/shap.d.ts +0 -657
  103. package/dist/shap/shap.d.ts.map +0 -1
  104. package/dist/shap/shap.js +0 -241
  105. package/dist/shap/shap.js.map +0 -1
  106. package/dist/simanneal/simanneal.d.ts.map +0 -1
  107. package/dist/simanneal/simanneal.js.map +0 -1
  108. package/dist/sklearn/sklearn.d.ts +0 -2691
  109. package/dist/sklearn/sklearn.d.ts.map +0 -1
  110. package/dist/sklearn/sklearn.js +0 -524
  111. package/dist/sklearn/sklearn.js.map +0 -1
  112. package/dist/torch/torch.d.ts +0 -1081
  113. package/dist/torch/torch.d.ts.map +0 -1
  114. package/dist/torch/torch.js.map +0 -1
  115. package/dist/tsconfig.tsbuildinfo +0 -1
  116. package/dist/types.d.ts +0 -80
  117. package/dist/types.d.ts.map +0 -1
  118. package/dist/types.js +0 -81
  119. package/dist/types.js.map +0 -1
  120. package/dist/xgboost/xgboost.d.ts +0 -504
  121. package/dist/xgboost/xgboost.d.ts.map +0 -1
  122. package/dist/xgboost/xgboost.js +0 -177
  123. package/dist/xgboost/xgboost.js.map +0 -1
@@ -0,0 +1 @@
1
+ {"version":3,"file":"lightning.d.ts","sourceRoot":"","sources":["../../../src/lightning/lightning.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,UAAU,EACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGrD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,0CAA0C;;IAE1C,2CAA2C;;QAEvC,yEAAyE;;;IAG7E,wDAAwD;;QAEpD,wBAAwB;;QAExB,iCAAiC;;;IAGrC,+DAA+D;;QAE3D,4CAA4C;;QAE5C,sCAAsC;;QAEtC,yDAAyD;;;EAG/D,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,QAAQ;;;EAGnB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,yBAAyB;IAClC,0CAA0C;;QAEtC,yBAAyB;;;IAG7B,+DAA+D;;QAE3D,iCAAiC;;QAEjC,oCAAoC;;QAEpC,iCAAiC;;;IAGrC,iEAAiE;;QAE7D,gDAAgD;;QAEhD,mCAAmC;;QAEnC,+BAA+B;;QAE/B,iDAAiD;;QAEjD,wCAAwC;;QAExC,8DAA8D;;;IAGlE,mEAAmE;;QAE/D,gDAAgD;;QAEhD,mCAAmC;;QAEnC,sBAAsB;;QAEtB,2BAA2B;;QAE3B,6BAA6B;;;;;QAE7B,iDAAiD;;QAEjD,+DAA+D;;QAE/D,8DAA8D;;;IAGlE,oEAAoE;;QAEhE,gDAAgD;;QAEhD,mCAAmC;;QAEnC,sBAAsB;;QAEtB,6DAA6D;;QAE7D,mCAAmC;;QAEnC,mDAAmD;;QAEnD,4CAA4C;;QAE5C,8DAA8D;;;EAGpE,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,0BAA0B,6DAGtC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,yBAAyB;;QAjFzB,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;IAmBlE,6CAA6C;;QAxH7C,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;IAsG7D,oCAAoC;;IAEpC,oCAAoC;;IAEpC,4CAA4C;;IAE5C,+BAA+B;;IAE/B,kCAAkC;;IAElC,6CAA6C;;IAE7C,kDAAkD;;IAElD,sCAAsC;;IAEtC,0CAA0C;;EAE5C,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,sBAAsB;;QAE3B,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;EAGhD,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,yBAAyB;;;YAnBrB,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAW9C,0BAA0B;;IAE1B,4BAA4B;;IAE5B,sCAAsC;;EAExC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,gBAAgB,8CAA+C,CAAC;AAE7E;;;;;GAKG;AACH,eAAO,MAAM,gBAAgB;IACzB,uDAAuD;;QAEnD,qEAAqE;;QAErE,mFAAmF;;;IAGvF,0CAA0C;;EAE5C,CAAC;AAMH;;;;;;;;;;GAUG;AACH,eAAO,MAAM,eAAe;IAhGxB,yBAAyB;;QAjFzB,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;IAmBlE,6CAA6C;;QAxH7C,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;IAsG7D,oCAAoC;;IAEpC,oCAAoC;;IAEpC,4CAA4C;;IAE5C,+BAA+B;;IAE/B,kCAAkC;;IAElC,6CAA6C;;IAE7C,kDAAkD;;IAElD,sCAAsC;;IAEtC,0CAA0C;;;IAkD1C,uDAAuD;;QAEnD,qEAAqE;;QAErE,mFAAmF;;;IAGvF,0CAA0C;;;IA7B1C,yBAAyB;;;YAnBrB,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAW9C,0BAA0B;;IAE1B,4BAA4B;;IAE5B,sCAAsC;;GA8CzC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,iBAAiB;;QAlFtB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;+IA4EjD,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB;;QA/FrB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;kDAyFjD,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB;;QA5GrB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;kDAsGjD,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,4BAA4B;;QA1HjC,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;yEAoHjD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B;IACpC,kCAAkC;;IAElC,gEAAgE;;IAEhE,+DAA+D;;EAEjE,CAAC;AAEH;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,2BAA2B;;QA1JhC,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;;IA0H9C,kCAAkC;;IAElC,gEAAgE;;IAEhE,+DAA+D;;2BAsBlE,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,cAAc;;QA1TvB,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;;QAiB7D,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;;;;;;;QAiBlE,yBAAyB;;YAjFzB,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;QAmBlE,6CAA6C;;YAxH7C,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;QAsG7D,oCAAoC;;QAEpC,oCAAoC;;QAEpC,4CAA4C;;QAE5C,+BAA+B;;QAE/B,kCAAkC;;QAElC,6CAA6C;;QAE7C,kDAAkD;;QAElD,sCAAsC;;QAEtC,0CAA0C;;;;QA4B1C,yBAAyB;;;gBAnBrB,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;QAW9C,0BAA0B;;QAE1B,4BAA4B;;QAE5B,sCAAsC;;;;;YAzBlC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;;;QA+B9C,uDAAuD;;YAEnD,qEAAqE;;YAErE,mFAAmF;;;QAGvF,0CAA0C;;;;QAoF1C,kCAAkC;;QAElC,gEAAgE;;QAEhE,+DAA+D;;;CA0CzD,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,SAAS;IAClB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;;QA3QH,yBAAyB;;YAjFzB,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;QAmBlE,6CAA6C;;YAxH7C,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;QAsG7D,oCAAoC;;QAEpC,oCAAoC;;QAEpC,4CAA4C;;QAE5C,+BAA+B;;QAE/B,kCAAkC;;QAElC,6CAA6C;;QAE7C,kDAAkD;;QAElD,sCAAsC;;QAEtC,0CAA0C;;;QAkD1C,uDAAuD;;YAEnD,qEAAqE;;YAErE,mFAAmF;;;QAGvF,0CAA0C;;;QA7B1C,yBAAyB;;;gBAnBrB,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;QAW9C,0BAA0B;;QAE1B,4BAA4B;;QAE5B,sCAAsC;;;IAwNtC;;;;;;;;;;;;;;;;;;OAkBG;;;YAnQC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IA4P9C;;;;;;;;;;;;;;;;;OAiBG;;;YAvRC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAgR9C;;;;;;;;;;;;;;;;;OAiBG;;;YA3SC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAoS9C;;;;;;;;;;;;;;;;;OAiBG;;;YA/TC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAwT9C;;;;;;;;;;;;;;;;;;OAkBG;;;YApVC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;QA0H9C,kCAAkC;;QAElC,gEAAgE;;QAEhE,+DAA+D;;;IA+M/D;;OAEG;;;YA5eH,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;;YAiB7D,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;;;;;;;YAiBlE,yBAAyB;;gBAjFzB,0CAA0C;;oBAEtC,yBAAyB;;;gBAG7B,+DAA+D;;oBAE3D,iCAAiC;;oBAEjC,oCAAoC;;oBAEpC,iCAAiC;;;gBAGrC,iEAAiE;;oBAE7D,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,+BAA+B;;oBAE/B,iDAAiD;;oBAEjD,wCAAwC;;oBAExC,8DAA8D;;;gBAGlE,mEAAmE;;oBAE/D,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,sBAAsB;;oBAEtB,2BAA2B;;oBAE3B,6BAA6B;;;;;oBAE7B,iDAAiD;;oBAEjD,+DAA+D;;oBAE/D,8DAA8D;;;gBAGlE,oEAAoE;;oBAEhE,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,sBAAsB;;oBAEtB,6DAA6D;;oBAE7D,mCAAmC;;oBAEnC,mDAAmD;;oBAEnD,4CAA4C;;oBAE5C,8DAA8D;;;;YAmBlE,6CAA6C;;gBAxH7C,0CAA0C;;gBAE1C,2CAA2C;;oBAEvC,yEAAyE;;;gBAG7E,wDAAwD;;oBAEpD,wBAAwB;;oBAExB,iCAAiC;;;gBAGrC,+DAA+D;;oBAE3D,4CAA4C;;oBAE5C,sCAAsC;;oBAEtC,yDAAyD;;;;YAsG7D,oCAAoC;;YAEpC,oCAAoC;;YAEpC,4CAA4C;;YAE5C,+BAA+B;;YAE/B,kCAAkC;;YAElC,6CAA6C;;YAE7C,kDAAkD;;YAElD,sCAAsC;;YAEtC,0CAA0C;;;;YA4B1C,yBAAyB;;;oBAnBrB,mDAAmD;;oBAEnD,sBAAsB;;oBAEtB,uBAAuB;;oBAEvB,wBAAwB;;oBAExB,kBAAkB;;oBAElB,0CAA0C;;;;YAW9C,0BAA0B;;YAE1B,4BAA4B;;YAE5B,sCAAsC;;;;;gBAzBlC,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;;;YA+B9C,uDAAuD;;gBAEnD,qEAAqE;;gBAErE,mFAAmF;;;YAGvF,0CAA0C;;;;YAoF1C,kCAAkC;;YAElC,gEAAgE;;YAEhE,+DAA+D;;;;CAmNzD,CAAC"}
@@ -0,0 +1,468 @@
1
+ /**
2
+ * Copyright (c) 2025 Elara AI Pty Ltd
3
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
4
+ */
5
+ /**
6
+ * Lightning platform functions for East.
7
+ *
8
+ * Provides production-grade neural network training using PyTorch Lightning.
9
+ * Supports regression, binary classification, multiclass classification,
10
+ * and multi-head categorical outputs.
11
+ *
12
+ * @packageDocumentation
13
+ */
14
+ import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, ArrayType, NullType, BooleanType, FunctionType, StringType, } from "@elaraai/east";
15
+ import { VectorType, MatrixType } from "../types.js";
16
+ // Re-export shared types
17
+ export { VectorType, MatrixType } from "../types.js";
18
+ // ===========================================
19
+ // Type Definitions
20
+ // ===========================================
21
+ /**
22
+ * Lightning output mode - determines loss function and output activation.
23
+ */
24
+ export const LightningOutputType = VariantType({
25
+ /** Regression: MSE loss, no activation */
26
+ regression: NullType,
27
+ /** Binary: BCE loss, sigmoid activation */
28
+ binary: StructType({
29
+ /** Optional per-position pos_weights for class imbalance [output_dim] */
30
+ pos_weight: OptionType(VectorType(FloatType)),
31
+ }),
32
+ /** Multiclass: CrossEntropy loss, softmax activation */
33
+ multiclass: StructType({
34
+ /** Number of classes */
35
+ n_classes: IntegerType,
36
+ /** Optional per-class weights */
37
+ class_weights: OptionType(VectorType(FloatType)),
38
+ }),
39
+ /** Multi-head categorical: N independent CrossEntropy heads */
40
+ multi_head: StructType({
41
+ /** Number of heads (e.g., 84 time slots) */
42
+ n_heads: IntegerType,
43
+ /** Classes per head (e.g., 4 bins) */
44
+ n_classes_per_head: IntegerType,
45
+ /** Optional class weights matrix (n_heads, n_classes) */
46
+ class_weights: OptionType(MatrixType(FloatType)),
47
+ }),
48
+ });
49
+ /**
50
+ * Cell type for sequential architectures.
51
+ */
52
+ export const CellType = VariantType({
53
+ lstm: NullType,
54
+ gru: NullType,
55
+ });
56
+ /**
57
+ * Lightning architecture type.
58
+ */
59
+ export const LightningArchitectureType = VariantType({
60
+ /** Simple MLP: input → hidden → output */
61
+ mlp: StructType({
62
+ /** Hidden layer sizes */
63
+ hidden_layers: ArrayType(IntegerType),
64
+ }),
65
+ /** Autoencoder: input → encoder → latent → decoder → output */
66
+ autoencoder: StructType({
67
+ /** Encoder hidden layer sizes */
68
+ encoder_layers: ArrayType(IntegerType),
69
+ /** Latent dimension (bottleneck) */
70
+ latent_dim: IntegerType,
71
+ /** Decoder hidden layer sizes */
72
+ decoder_layers: ArrayType(IntegerType),
73
+ }),
74
+ /** Conv1D: 1D convolutional autoencoder for temporal patterns */
75
+ conv1d: StructType({
76
+ /** Number of channels (e.g., additive types) */
77
+ n_channels: IntegerType,
78
+ /** Sequence length (e.g., days) */
79
+ sequence_length: IntegerType,
80
+ /** Conv layer channel sizes */
81
+ conv_channels: ArrayType(IntegerType),
82
+ /** Kernel size for convolutions (must be odd) */
83
+ kernel_size: IntegerType,
84
+ /** Latent dimension after flattening */
85
+ latent_dim: IntegerType,
86
+ /** Optional condition dimension for conditional generation */
87
+ condition_dim: OptionType(IntegerType),
88
+ }),
89
+ /** Sequential: LSTM/GRU autoencoder for long-range dependencies */
90
+ sequential: StructType({
91
+ /** Number of channels (e.g., additive types) */
92
+ n_channels: IntegerType,
93
+ /** Sequence length (e.g., days) */
94
+ sequence_length: IntegerType,
95
+ /** RNN hidden size */
96
+ hidden_size: IntegerType,
97
+ /** Number of RNN layers */
98
+ n_layers: IntegerType,
99
+ /** Cell type: lstm or gru */
100
+ cell_type: CellType,
101
+ /** Latent dimension (from final hidden state) */
102
+ latent_dim: IntegerType,
103
+ /** Bidirectional encoder (decoder is always unidirectional) */
104
+ bidirectional: BooleanType,
105
+ /** Optional condition dimension for conditional generation */
106
+ condition_dim: OptionType(IntegerType),
107
+ }),
108
+ /** Transformer: attention-based autoencoder for complex patterns */
109
+ transformer: StructType({
110
+ /** Number of channels (e.g., additive types) */
111
+ n_channels: IntegerType,
112
+ /** Sequence length (e.g., days) */
113
+ sequence_length: IntegerType,
114
+ /** Model dimension */
115
+ d_model: IntegerType,
116
+ /** Number of attention heads (must divide d_model evenly) */
117
+ n_attention_heads: IntegerType,
118
+ /** Number of transformer layers */
119
+ n_layers: IntegerType,
120
+ /** Feedforward dimension (default: 4 * d_model) */
121
+ d_ff: OptionType(IntegerType),
122
+ /** Latent dimension (mean pooled output) */
123
+ latent_dim: IntegerType,
124
+ /** Optional condition dimension for conditional generation */
125
+ condition_dim: OptionType(IntegerType),
126
+ }),
127
+ });
128
+ /**
129
+ * Epoch callback function type: (epoch, train_loss, val_loss) -> void
130
+ */
131
+ export const LightningEpochCallbackType = FunctionType([IntegerType, FloatType, FloatType], NullType);
132
+ /**
133
+ * Lightning training configuration.
134
+ */
135
+ export const LightningConfigType = StructType({
136
+ /** Model architecture */
137
+ architecture: LightningArchitectureType,
138
+ /** Output mode (determines loss function) */
139
+ output: LightningOutputType,
140
+ /** Learning rate (default: 1e-3) */
141
+ learning_rate: OptionType(FloatType),
142
+ /** Maximum epochs (default: 100) */
143
+ max_epochs: OptionType(IntegerType),
144
+ /** Early stopping patience (default: 10) */
145
+ patience: OptionType(IntegerType),
146
+ /** Batch size (default: 32) */
147
+ batch_size: OptionType(IntegerType),
148
+ /** Dropout rate (default: 0.1) */
149
+ dropout: OptionType(FloatType),
150
+ /** Gradient clipping value (default: 1.0) */
151
+ gradient_clip: OptionType(FloatType),
152
+ /** L2 regularization weight decay (default: 0) */
153
+ weight_decay: OptionType(FloatType),
154
+ /** Random seed for reproducibility */
155
+ random_state: OptionType(IntegerType),
156
+ /** Optional callback called each epoch */
157
+ epoch_callback: OptionType(LightningEpochCallbackType),
158
+ });
159
+ /**
160
+ * Lightning model blob structure.
161
+ */
162
+ export const LightningModelBlobType = VariantType({
163
+ lightning: StructType({
164
+ /** Serialized model data (state_dict + hparams) */
165
+ data: BlobType,
166
+ /** Input dimension */
167
+ n_features: IntegerType,
168
+ /** Output dimension */
169
+ output_dim: IntegerType,
170
+ /** Architecture type */
171
+ architecture_type: StringType,
172
+ /** Output type */
173
+ output_type: StringType,
174
+ /** Latent dimension (autoencoder only) */
175
+ latent_dim: OptionType(IntegerType),
176
+ }),
177
+ });
178
+ /**
179
+ * Lightning training result.
180
+ */
181
+ export const LightningResultType = StructType({
182
+ /** Trained model blob */
183
+ model: LightningModelBlobType,
184
+ /** Final training loss */
185
+ train_loss: FloatType,
186
+ /** Final validation loss */
187
+ val_loss: FloatType,
188
+ /** Best epoch (for early stopping) */
189
+ best_epoch: IntegerType,
190
+ });
191
+ /**
192
+ * 3D boolean tensor for masks: (n_samples, n_heads, n_classes)
193
+ */
194
+ export const Tensor3DBoolType = ArrayType(ArrayType(ArrayType(BooleanType)));
195
+ /**
196
+ * Group-based weights for per-sample class weighting.
197
+ *
198
+ * Instead of per-sample weights (memory-intensive), samples belong to discrete
199
+ * groups (e.g., grades) with different weight configurations per group.
200
+ */
201
+ export const GroupWeightsType = StructType({
202
+ /** Weights per group - shape depends on output type */
203
+ weights: VariantType({
204
+ /** For binary: pos_weight vector per group [n_groups][output_dim] */
205
+ binary: ArrayType(ArrayType(FloatType)),
206
+ /** For multi_head: class_weight matrix per group [n_groups][n_heads][n_classes] */
207
+ multi_head: ArrayType(ArrayType(ArrayType(FloatType))),
208
+ }),
209
+ /** Group index per sample: [n_samples] */
210
+ sample_groups: ArrayType(IntegerType),
211
+ });
212
+ // ===========================================
213
+ // Platform Functions
214
+ // ===========================================
215
+ /**
216
+ * Train a Lightning model.
217
+ *
218
+ * @param X - Input features matrix (n_samples, n_features)
219
+ * @param y - Target matrix (n_samples, output_dim)
220
+ * @param config - Training configuration
221
+ * @param masks - Optional 3D boolean masks (n_samples, n_heads, n_classes)
222
+ * @param group_weights - Optional group-based weights for per-sample weighting
223
+ * @param conditions - Optional condition matrix for conditional generation (n_samples, condition_dim)
224
+ * @returns Training result with model blob and metrics
225
+ */
226
+ export const lightning_train = East.platform("lightning_train", [MatrixType(FloatType), MatrixType(FloatType), LightningConfigType, OptionType(Tensor3DBoolType), OptionType(GroupWeightsType), OptionType(MatrixType(FloatType))], LightningResultType);
227
+ /**
228
+ * Predict using a Lightning model.
229
+ *
230
+ * @param model - Trained model blob
231
+ * @param X - Input features matrix (n_samples, n_features)
232
+ * @param masks - Optional 3D boolean masks for inference
233
+ * @param conditions - Optional condition matrix for conditional models (n_samples, condition_dim)
234
+ * @returns Predicted probabilities matrix (n_samples, output_dim)
235
+ */
236
+ export const lightning_predict = East.platform("lightning_predict", [LightningModelBlobType, MatrixType(FloatType), OptionType(Tensor3DBoolType), OptionType(MatrixType(FloatType))], MatrixType(FloatType));
237
+ /**
238
+ * Encode input to latent space (autoencoder only).
239
+ *
240
+ * @param model - Trained autoencoder model blob
241
+ * @param X - Input features matrix (n_samples, n_features)
242
+ * @returns Latent embeddings matrix (n_samples, latent_dim)
243
+ */
244
+ export const lightning_encode = East.platform("lightning_encode", [LightningModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
245
+ /**
246
+ * Decode latent to output (autoencoder only).
247
+ *
248
+ * @param model - Trained autoencoder model blob
249
+ * @param z - Latent embeddings matrix (n_samples, latent_dim)
250
+ * @returns Decoded output matrix (n_samples, output_dim)
251
+ */
252
+ export const lightning_decode = East.platform("lightning_decode", [LightningModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
253
+ /**
254
+ * Decode latent to output with condition (temporal architectures with condition_dim).
255
+ *
256
+ * @param model - Trained model with condition_dim set
257
+ * @param z - Latent embeddings matrix (n_samples, latent_dim)
258
+ * @param condition - Condition vectors (n_samples, condition_dim)
259
+ * @returns Decoded output matrix (n_samples, output_dim)
260
+ */
261
+ export const lightning_decode_conditional = East.platform("lightning_decode_conditional", [LightningModelBlobType, MatrixType(FloatType), MatrixType(FloatType)], MatrixType(FloatType));
262
+ /**
263
+ * Configuration for autoregressive sequence generation.
264
+ */
265
+ export const LightningGenerateConfigType = StructType({
266
+ /** Number of steps to generate */
267
+ n_steps: IntegerType,
268
+ /** Sampling temperature: 0.0 = argmax, > 0 = scaled sampling */
269
+ temperature: FloatType,
270
+ /** If true, return probabilities. If false, return samples. */
271
+ return_probs: BooleanType,
272
+ });
273
+ /**
274
+ * Generate sequence autoregressively from a sequential model.
275
+ *
276
+ * Shapes:
277
+ * - prefix: (n_prefix_steps, n_channels) - partial history to continue from, can be empty []
278
+ * - condition: (1, condition_dim) - conditioning features, or none
279
+ * - returns: (n_steps, n_channels) - generated timesteps only (not including prefix)
280
+ *
281
+ * @param model - Trained sequential model blob
282
+ * @param prefix - Partial history to continue from
283
+ * @param condition - Optional conditioning features
284
+ * @param config - Generation configuration
285
+ * @returns Generated sequence matrix
286
+ */
287
+ export const lightning_generate_sequence = East.platform("lightning_generate_sequence", [LightningModelBlobType, MatrixType(FloatType), OptionType(MatrixType(FloatType)), LightningGenerateConfigType], MatrixType(FloatType));
288
+ // ===========================================
289
+ // Grouped Export
290
+ // ===========================================
291
+ /**
292
+ * Lightning types namespace.
293
+ */
294
+ export const LightningTypes = {
295
+ OutputType: LightningOutputType,
296
+ ArchitectureType: LightningArchitectureType,
297
+ CellType,
298
+ EpochCallbackType: LightningEpochCallbackType,
299
+ ConfigType: LightningConfigType,
300
+ ResultType: LightningResultType,
301
+ ModelBlobType: LightningModelBlobType,
302
+ Tensor3DBoolType,
303
+ GroupWeightsType,
304
+ GenerateConfigType: LightningGenerateConfigType,
305
+ };
306
+ /**
307
+ * Lightning platform functions namespace.
308
+ *
309
+ * Provides production-grade neural network training using PyTorch Lightning.
310
+ *
311
+ * @example
312
+ * ```typescript
313
+ * const result = Lightning.train(X, y, {
314
+ * architecture: variant("autoencoder", {
315
+ * encoder_layers: [64n],
316
+ * latent_dim: 16n,
317
+ * decoder_layers: [64n],
318
+ * }),
319
+ * output: variant("multi_head", {
320
+ * n_heads: 84n,
321
+ * n_classes_per_head: 4n,
322
+ * class_weights: variant("none", null),
323
+ * }),
324
+ * }, variant("none", null));
325
+ *
326
+ * const embeddings = Lightning.encode(result.model, X);
327
+ * const predictions = Lightning.predict(result.model, X, variant("none", null));
328
+ * ```
329
+ */
330
+ export const Lightning = {
331
+ /**
332
+ * Train a Lightning model.
333
+ *
334
+ * Trains a neural network using PyTorch Lightning with early stopping,
335
+ * gradient clipping, and optional epoch callbacks.
336
+ *
337
+ * @example
338
+ * ```ts
339
+ * import { East, FloatType, MatrixType, variant } from "@elaraai/east";
340
+ * import { Lightning, LightningConfigType } from "@elaraai/east-py-datascience";
341
+ *
342
+ * const train = East.function([], Lightning.Types.ResultType, ($) => {
343
+ * const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
344
+ * const y = $.let([[1.0, 0.0], [0.0, 1.0], [1.0, 0.0], [0.0, 1.0]]);
345
+ * const config = $.let({
346
+ * architecture: variant("autoencoder", {
347
+ * encoder_layers: new BigInt64Array([16n]),
348
+ * latent_dim: 4n,
349
+ * decoder_layers: new BigInt64Array([16n]),
350
+ * }),
351
+ * output: variant("regression", null),
352
+ * max_epochs: variant("some", 50n),
353
+ * batch_size: variant("some", 4n),
354
+ * learning_rate: variant("some", 0.001),
355
+ * patience: variant("some", 10n),
356
+ * seed: variant("some", 42n),
357
+ * epoch_callback: variant("none", null),
358
+ * gradient_clip_val: variant("none", null),
359
+ * condition_dim: variant("none", null),
360
+ * }, LightningConfigType);
361
+ * return $.return(Lightning.train(X, y, config, variant("none", null), variant("none", null), variant("none", null)));
362
+ * });
363
+ * ```
364
+ */
365
+ train: lightning_train,
366
+ /**
367
+ * Predict using a trained Lightning model.
368
+ *
369
+ * Returns predictions with optional mask support for multi-head outputs.
370
+ *
371
+ * @example
372
+ * ```ts
373
+ * import { East, FloatType, MatrixType, variant } from "@elaraai/east";
374
+ * import { Lightning } from "@elaraai/east-py-datascience";
375
+ *
376
+ * const predictFn = East.function(
377
+ * [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
378
+ * MatrixType(FloatType),
379
+ * ($, model, X) => {
380
+ * return $.return(Lightning.predict(model, X, variant("none", null), variant("none", null)));
381
+ * }
382
+ * );
383
+ * ```
384
+ */
385
+ predict: lightning_predict,
386
+ /**
387
+ * Encode inputs to latent space (autoencoder only).
388
+ *
389
+ * @example
390
+ * ```ts
391
+ * import { East, FloatType, MatrixType } from "@elaraai/east";
392
+ * import { Lightning } from "@elaraai/east-py-datascience";
393
+ *
394
+ * const encodeFn = East.function(
395
+ * [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
396
+ * MatrixType(FloatType),
397
+ * ($, model, X) => {
398
+ * // Returns (n_samples x latent_dim) embeddings
399
+ * return $.return(Lightning.encode(model, X));
400
+ * }
401
+ * );
402
+ * ```
403
+ */
404
+ encode: lightning_encode,
405
+ /**
406
+ * Decode latent embeddings to output space (autoencoder only).
407
+ *
408
+ * @example
409
+ * ```ts
410
+ * import { East, FloatType, MatrixType } from "@elaraai/east";
411
+ * import { Lightning } from "@elaraai/east-py-datascience";
412
+ *
413
+ * const decodeFn = East.function(
414
+ * [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
415
+ * MatrixType(FloatType),
416
+ * ($, model, z) => {
417
+ * // z is (n_samples x latent_dim), returns (n_samples x output_dim)
418
+ * return $.return(Lightning.decode(model, z));
419
+ * }
420
+ * );
421
+ * ```
422
+ */
423
+ decode: lightning_decode,
424
+ /**
425
+ * Decode latent embeddings with condition vector (temporal architectures).
426
+ *
427
+ * @example
428
+ * ```ts
429
+ * import { East, FloatType, MatrixType } from "@elaraai/east";
430
+ * import { Lightning } from "@elaraai/east-py-datascience";
431
+ *
432
+ * const decodeFn = East.function(
433
+ * [Lightning.Types.ModelBlobType, MatrixType(FloatType), MatrixType(FloatType)],
434
+ * MatrixType(FloatType),
435
+ * ($, model, z, condition) => {
436
+ * // z: (n_samples, latent_dim), condition: (n_samples, condition_dim)
437
+ * return $.return(Lightning.decodeConditional(model, z, condition));
438
+ * }
439
+ * );
440
+ * ```
441
+ */
442
+ decodeConditional: lightning_decode_conditional,
443
+ /**
444
+ * Generate sequence autoregressively from a sequential model.
445
+ *
446
+ * @example
447
+ * ```ts
448
+ * import { East, FloatType, IntegerType, BooleanType, MatrixType, variant } from "@elaraai/east";
449
+ * import { Lightning } from "@elaraai/east-py-datascience";
450
+ *
451
+ * const generateFn = East.function(
452
+ * [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
453
+ * MatrixType(FloatType),
454
+ * ($, model, prefix) => {
455
+ * const config = $.let({ n_steps: 10n, temperature: 1.0, return_probs: false });
456
+ * // prefix: partial history, condition: none
457
+ * return $.return(Lightning.generateSequence(model, prefix, variant("none", null), config));
458
+ * }
459
+ * );
460
+ * ```
461
+ */
462
+ generateSequence: lightning_generate_sequence,
463
+ /**
464
+ * Type definitions for Lightning functions.
465
+ */
466
+ Types: LightningTypes,
467
+ };
468
+ //# sourceMappingURL=lightning.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"lightning.js","sourceRoot":"","sources":["../../../src/lightning/lightning.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,UAAU,GACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yBAAyB;AACzB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,8CAA8C;AAC9C,mBAAmB;AACnB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;IAC3C,0CAA0C;IAC1C,UAAU,EAAE,QAAQ;IACpB,2CAA2C;IAC3C,MAAM,EAAE,UAAU,CAAC;QACf,yEAAyE;QACzE,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KAChD,CAAC;IACF,wDAAwD;IACxD,UAAU,EAAE,UAAU,CAAC;QACnB,wBAAwB;QACxB,SAAS,EAAE,WAAW;QACtB,iCAAiC;QACjC,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KACnD,CAAC;IACF,+DAA+D;IAC/D,UAAU,EAAE,UAAU,CAAC;QACnB,4CAA4C;QAC5C,OAAO,EAAE,WAAW;QACpB,sCAAsC;QACtC,kBAAkB,EAAE,WAAW;QAC/B,yDAAyD;QACzD,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KACnD,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC;IAChC,IAAI,EAAE,QAAQ;IACd,GAAG,EAAE,QAAQ;CAChB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,WAAW,CAAC;IACjD,0CAA0C;IAC1C,GAAG,EAAE,UAAU,CAAC;QACZ,yBAAyB;QACzB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;KACxC,CAAC;IACF,+DAA+D;IAC/D,WAAW,EAAE,UAAU,CAAC;QACpB,iCAAiC;QACjC,cAAc,EAAE,SAAS,CAAC,WAAW,CAAC;QACtC,oCAAoC;QACpC,UAAU,EAAE,WAAW;QACvB,iCAAiC;QACjC,cAAc,EAAE,SAAS,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,iEAAiE;IACjE,MAAM,EAAE,UAAU,CAAC;QACf,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,+BAA+B;QAC/B,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;QACrC,iDAAiD;QACjD,WAAW,EAAE,WAAW;QACxB,wCAAwC;QACxC,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,mEAAmE;IACnE,UAAU,EAAE,UAAU,CAAC;QACnB,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,sBAAsB;QACtB,WAAW,EAAE,WAAW;QACxB,2BAA2B;QAC3B,QAAQ,EAAE,WAAW;QACrB,6BAA6B;QAC7B,SAAS,EAAE,QAAQ;QACnB,iDAAiD;QACjD,UAAU,EAAE,WAAW;QACvB,+DAA+D;QAC/D,aAAa,EAAE,WAAW;QAC1B,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,oEAAoE;IACpE,WAAW,EAAE,UAAU,CAAC;QACpB,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,sBAAsB;QACtB,OAAO,EAAE,WAAW;QACpB,6DAA6D;QAC7D,iBAAiB,EAAE,WAAW;QAC9B,mCAAmC;QACnC,QAAQ,EAAE,WAAW;QACrB,mDAAmD;QACnD,IAAI,EAAE,UAAU,CAAC,WAAW,CAAC;QAC7B,4CAA4C;QAC5C,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,YAAY,CAClD,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,EACnC,QAAQ,CACX,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,yBAAyB;IACzB,YAAY,EAAE,yBAAyB;IACvC,6CAA6C;IAC7C,MAAM,EAAE,mBAAmB;IAC3B,oCAAoC;IACpC,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,oCAAoC;IACpC,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,4CAA4C;IAC5C,QAAQ,EAAE,UAAU,CAAC,WAAW,CAAC;IACjC,+BAA+B;IAC/B,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,kCAAkC;IAClC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC;IAC9B,6CAA6C;IAC7C,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,kDAAkD;IAClD,YAAY,EAAE,UAAU,CAAC,SAAS,CAAC;IACnC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,0CAA0C;IAC1C,cAAc,EAAE,UAAU,CAAC,0BAA0B,CAAC;CACzD,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,WAAW,CAAC;IAC9C,SAAS,EAAE,UAAU,CAAC;QAClB,mDAAmD;QACnD,IAAI,EAAE,QAAQ;QACd,sBAAsB;QACtB,UAAU,EAAE,WAAW;QACvB,uBAAuB;QACvB,UAAU,EAAE,WAAW;QACvB,wBAAwB;QACxB,iBAAiB,EAAE,UAAU;QAC7B,kBAAkB;QAClB,WAAW,EAAE,UAAU;QACvB,0CAA0C;QAC1C,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;KACtC,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,yBAAyB;IACzB,KAAK,EAAE,sBAAsB;IAC7B,0BAA0B;IAC1B,UAAU,EAAE,SAAS;IACrB,4BAA4B;IAC5B,QAAQ,EAAE,SAAS;IACnB,sCAAsC;IACtC,UAAU,EAAE,WAAW;CAC1B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;AAE7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;IACvC,uDAAuD;IACvD,OAAO,EAAE,WAAW,CAAC;QACjB,qEAAqE;QACrE,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACvC,mFAAmF;QACnF,UAAU,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;KACzD,CAAC;IACF,0CAA0C;IAC1C,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,8CAA8C;AAC9C,qBAAqB;AACrB,8CAA8C;AAE9C;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAClK,mBAAmB,CACtB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAC1C,mBAAmB,EACnB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAChH,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC/C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC/C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,IAAI,CAAC,QAAQ,CACrD,8BAA8B,EAC9B,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EACtE,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,UAAU,CAAC;IAClD,kCAAkC;IAClC,OAAO,EAAE,WAAW;IACpB,gEAAgE;IAChE,WAAW,EAAE,SAAS;IACtB,+DAA+D;IAC/D,YAAY,EAAE,WAAW;CAC5B,CAAC,CAAC;AAEH;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,IAAI,CAAC,QAAQ,CACpD,6BAA6B,EAC7B,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,2BAA2B,CAAC,EAC/G,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF,8CAA8C;AAC9C,iBAAiB;AACjB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG;IAC1B,UAAU,EAAE,mBAAmB;IAC/B,gBAAgB,EAAE,yBAAyB;IAC3C,QAAQ;IACR,iBAAiB,EAAE,0BAA0B;IAC7C,UAAU,EAAE,mBAAmB;IAC/B,UAAU,EAAE,mBAAmB;IAC/B,aAAa,EAAE,sBAAsB;IACrC,gBAAgB;IAChB,gBAAgB;IAChB,kBAAkB,EAAE,2BAA2B;CACzC,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG;IACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,KAAK,EAAE,eAAe;IAEtB;;;;;;;;;;;;;;;;;;OAkBG;IACH,OAAO,EAAE,iBAAiB;IAE1B;;;;;;;;;;;;;;;;;OAiBG;IACH,MAAM,EAAE,gBAAgB;IAExB;;;;;;;;;;;;;;;;;OAiBG;IACH,MAAM,EAAE,gBAAgB;IAExB;;;;;;;;;;;;;;;;;OAiBG;IACH,iBAAiB,EAAE,4BAA4B;IAE/C;;;;;;;;;;;;;;;;;;OAkBG;IACH,gBAAgB,EAAE,2BAA2B;IAE7C;;OAEG;IACH,KAAK,EAAE,cAAc;CACf,CAAC"}