Skip to content
/ simple-jax-to-tensorrt-example Public

A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.

License

Apache-2.0 license
1 star 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

ChristianOrr/simple-jax-to-tensorrt-example

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits

common

common

 
 

data

data

 
 

.gitignore

.gitignore

 
 

CMakeLists.txt

CMakeLists.txt

 
 

LICENSE

LICENSE

 
 

README.md

README.md

 
 

jax_mnist_haiku.ipynb

jax_mnist_haiku.ipynb

 
 

jax_mnist_mlp.ipynb

jax_mnist_mlp.ipynb

 
 

mnist_inference.cpp

mnist_inference.cpp

 
 

Repository files navigation

simple-jax-to-tensorrt-example

A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.

About

A MNIST model is defined and trained in Jax, then converted to TensorRT for inference.

Topics

tensorrt jax

Resources

Readme

License

Apache-2.0 license
Activity

Stars

1 star

Watchers

2 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages