batch_rng_split(k)
|
Vectorized version of <lambda>. |
fit_morse_potential(d, E[, p0])
|
Fits a morse potential to given energy data. |
getCodeVersion()
|
Determine the current git commit by calling 'git log -1'. |
get_distance_matrix(r_el)
|
Compute distance matrix omitting the main diagonal (i.e. |
get_el_ion_distance_matrix(r_el, R_ion)
|
- param r_el
shape [N_batch x n_el x 3]
|
get_extent_for_imshow(x, y)
|
|
get_from_devices(data)
|
|
get_full_distance_matrix(r_el)
|
- param r_el
shape [n_el x 3]
|
get_ion_pos_string(R)
|
|
get_number_of_params(params)
|
|
get_params_filter_func(module_names)
|
|
init_multi_host_on_slurm()
|
|
is_equal_across_devices(data)
|
|
merge_from_devices(x)
|
|
merge_params(params_old, params_new)
|
|
morse_potential(r, E, a, Rb, E0)
|
Returns the Morse potential \(E_0 + E (exp{-2x} - 2exp{-x})\) where \(x = a (r-R_b)\) |
replicate_across_devices(data)
|
|
save_xyz_file(fname, R, Z[, comment])
|
Units for R are expected to be Bohr and will be translated to Angstrom in output |
setup_job_dir(parent_dir, name)
|
|
split_params(params, module_names)
|
|