Model Library

  

SciAI model library provides a wide variety of models that are frequently used and cited in scientific computation. The following table summarizes the current available neural networks and their corresponding domains.

Domain	Network	MindSpore Implementation and Parameters	Ascend	GPU
General Physics	auq_pinns	link	✅	✅
General Physics	cpinns	link	✅	✅
General Physics	deep_hpms	link	✅	✅
General Physics	deep_ritz	link	✅	✅
General Physics	deepbsde	link		✅
General Physics	deeponet	link	✅	✅
General Physics	dgm	link	✅	✅
General Physics	fbsnns	link	✅	✅
General Physics	fpinns	link	✅	✅
General Physics	gradient_pathologies_pinns	link	✅	✅
General Physics	hp_vpinns	link	✅	✅
General Physics	laaf	link	✅	✅
General Physics	mgnet	link	✅	✅
General Physics	multiscale_pinns	link	✅	✅
General Physics	pfnn	link		✅
General Physics	phygeonet	link	✅	✅
General Physics	pi_deeponet	link		✅
General Physics	pinns	link		✅
General Physics	pinns_ntk	link	✅	✅
General Physics	ppinns	link	✅	✅
General Physics	xpinns	link	✅	✅
Hamiltonian Systems	sympnets	link	✅	✅
Fluid Dynamic	hfm	link	✅	✅
Fluid Dynamic	label_free_dnn_surrogate	link	✅	✅
Fluid Dynamic	nsf_nets	link	✅	✅
Fluid Dynamic	*burgers_fno	link	✅	✅
Fluid Dynamic	*burgers_kno	link	✅	✅
Fluid Dynamic	*navier_stokes_fno	link	✅	✅
Fluid Dynamic	*navier_stokes_kno	link	✅	✅
Fluid Dynamic	*navier_stokes_3d_fno	link	✅	✅
Fluid Dynamic	*pde_net	link	✅	✅
Fluid Dynamic	*percnn	link	✅	✅
Elastodynamics	pinn_elastodynamics	link	✅	✅
Thermodynamics	pinn_heattransfer	link	✅	✅
Meteorology	enso	link	✅	✅
Geology	inversion_net	link	✅	✅
Geology	pinn_helmholtz	link	✅	✅
Oceanic Physics	ocean_model	link		✅
Oceanic Physics	pinns_swe	link	✅	✅
Electromagnetism	maxwell_net	link	✅	✅
Electromagnetism	*AD_FDTD_invert_f	link		✅
Electromagnetism	*AD_FDTD_microstrip_filter	link		✅
Electromagnetism	*AD_FDTD_inverse	link		✅
Electromagnetism	*frequency_domain_maxwell	link	✅	✅
Electromagnetism	*frequency_domain_maxwell_3D_dielectric_slab	link	✅	✅
Electromagnetism	*frequency_domain_maxwell_3D_waveguide_cavity	link	✅	✅
Electromagnetism	*meta_auto_decoder	link	✅	✅
Electromagnetism	*pinn_fwi	link	✅	✅
Electromagnetism	*SED_ANN	link	✅	✅
Electromagnetism	*time_domain_maxwell	link	✅	✅
Electromagnetism	*metasurface_holograms	link	✅	✅
Biology	*MEGA-Fold	link (inference) link (training)	✅	✅
Biology	*MEGA-EvoGen	link	✅	✅
Biology	*MEGA-Assessment	link (inference) link (training)	✅	✅
Biology	*ColabDesign	link	✅	✅
Biology	*DeepFRI	link	✅	✅
Biology	*Multimer	link	✅	✅
Biology	*ProteinMPNN	link	✅	✅
Biology	*UFold	link	✅	✅
Biology	*esm-if1	link	✅	✅
Biology	*esm2	link	✅	✅
Biology	*grover	link	✅	✅

Note: the "*" in the model names indicates that these models have already been released at an earlier time by MindSpore and MindScience.