Merge pull request #66 from actuarialopensource/compiled-model

Compiled model

Author: MatthewCaseres

…ithub/workflows/BasicTerm_ME_PyTorch.yml .github/workflows/BasicTerm_ME_python.yml.github/workflows/BasicTerm_ME_PyTorch.yml renamed to .github/workflows/BasicTerm_ME_python.yml .github/workflows/BasicTerm_ME_PyTorch.yml renamed to .github/workflows/BasicTerm_ME_python.yml

@@ -1,10 +1,10 @@
-name: basicterm_me_pytorch
+name: basicterm_me_python
 
 on:
   workflow_dispatch: 
   push:
     paths:
-      - 'containers/BasicTerm_ME_heavylight/**'
+      - 'containers/BasicTerm_ME_python/**'
 
 jobs:
   docker:
@@ -28,6 +28,6 @@ jobs:
         name: Build and push
         uses: docker/build-push-action@v5
         with:
-          context: "{{defaultContext}}:containers/BasicTerm_ME_heavylight"
+          context: "{{defaultContext}}:containers/BasicTerm_ME_python"
           push: true
-          tags: actuarial/basicterm_me_pytorch:latest
+          tags: actuarial/basicterm_me_python:latest

README.md

@@ -8,9 +8,10 @@ We currently only have 1 benchmark, we are working to expand the benchmarks. Ope
 
 Time measurement is currently the best of 3 runs.
 
-| benchmark     | recursive | container |A100-SXM4-40GB | H100-SXM5-80GB |
+| benchmark     | classification | container |A100-SXM4-40GB | H100-SXM5-80GB |
 |---------------|-|-|----------------|----------------|
-| BasicTerm_ME 100 Million | Yes | [link](https://hub.docker.com/repository/docker/actuarial/basicterm_me_pytorch/general) | 15.8194s        | 7.1820s         |
+| BasicTerm_ME 100 Million | recursive PyTorch | [link](https://hub.docker.com/repository/docker/actuarial/basicterm_me_python/general) | 15.8284s        | 7.205s         |
+| BasicTerm_ME 100 Million | compiled iterative JAX | [link](https://hub.docker.com/repository/docker/actuarial/basicterm_me_python/general) | 3.448s        | 1.551s         |
 
 
 ### Notes