lkhphuc commited on
Commit
de0a610
1 Parent(s): 2615464

show jax device

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,4 +1,7 @@
1
  import streamlit as st
 
2
 
3
  x = st.slider('Select a value')
4
  st.write(x, 'squared is', x * x)
 
 
1
  import streamlit as st
2
+ import jax
3
 
4
  x = st.slider('Select a value')
5
  st.write(x, 'squared is', x * x)
6
+
7
+ st.write(jax.device_count())