jaxrts
Contents:
Installation
Usage
Example gallery
Development and Poetry
Bibliography
The
jaxrts
module
jaxrts
jaxrts.absorption_edges_ionized_ions_data
jaxrts.analysis
jaxrts.bound_free
jaxrts.ee_localfieldcorrections
jaxrts.ee_localfieldcorrections_dornheim_2021
jaxrts.elements
jaxrts.experimental
jaxrts.experimental.SiiNN
jaxrts.form_factors
jaxrts.free_bound
jaxrts.free_free
jaxrts.helpers
jaxrts.hnc_potentials
jaxrts.hypernetted_chain
jaxrts.instrument_function
jaxrts.ion_feature
jaxrts.ipd
jaxrts.literature
jaxrts.math
jaxrts.models
jaxrts.plasma_physics
jaxrts.plasmastate
jaxrts.saha
jaxrts.saving
jaxrts.setup
jaxrts.static_structure_factors
jaxrts.units
jaxrts
The
jaxrts
module
jaxrts
jaxrts.experimental
jaxrts.experimental.SiiNN
jaxrts.experimental.SiiNN.set_sharding
View page source
jaxrts.experimental.SiiNN.set_sharding
jaxrts.experimental.SiiNN.
set_sharding
(
x
:
ShapeDtypeStruct
)
→
ShapeDtypeStruct
[source]