Struct RayIntegration 

pub struct RayIntegration { /* private fields */ }

Ray compatibility integration

Implementations

impl RayIntegration

pub fn new(config: RayConfig) -> Self

Create a new Ray integration

pub fn from_file<P: AsRef<Path>>(path: P) -> TorshResult<Self>

Load configuration from JSON file

pub fn initialize( &mut self, rank: u32, world_size: u32, local_rank: u32, local_size: u32, ) -> TorshResult<()>

Initialize Ray integration

pub fn to_elastic_config(&self) -> TorshResult<Option<ElasticConfig>>

Convert Ray config to ToRSh elastic config

pub fn config(&self) -> &RayConfig

Get the current configuration

pub fn stats(&self) -> &RayStats

Get current statistics

pub fn is_initialized(&self) -> bool

Check if Ray integration is initialized

pub fn rank(&self) -> u32

Get current rank

pub fn world_size(&self) -> u32

Get world size

pub fn local_rank(&self) -> u32

Get local rank

pub fn local_size(&self) -> u32

Get local size

pub fn is_ray_session_active(&self) -> bool

Check if Ray session is active

pub fn run_training( &mut self, train_func_name: &str, num_epochs: u32, ) -> TorshResult<()>

Simulate Ray Train run

pub fn run_tuning(&mut self, tune_config_name: &str) -> TorshResult<()>

Simulate Ray Tune run

pub fn shutdown(&mut self) -> TorshResult<()>

Shutdown Ray integration

pub fn default_config() -> RayConfig

Create a default Ray configuration

pub fn config_with_tune( num_samples: u32, search_alg: RaySearchAlgorithm, ) -> RayConfig

Create a configuration for hyperparameter tuning

pub fn config_with_serve(num_replicas: u32) -> RayConfig

Create a configuration for model serving