graphax.examples package

Submodules

graphax.examples.deep_learning module

graphax.examples.deep_learning.Encoder(x, y, WQ1, WQ2, WK1, WK2, WV1, WV2, W1, W2, b1, b2, gamma0, beta0, gamma1, beta1)[source]

graphax.examples.deep_learning.EncoderDecoder(x, y, WQ1, WQ2, WQ3, WK1, WK2, WK3, WV1, WV2, WV3, W1, W2, b1, b2, gamma0, beta0, gamma1, beta1, gamma2, beta2)[source]

graphax.examples.deep_learning.Perceptron(x, y, W1, b1, W2, b2, gamma, beta)[source]

graphax.examples.deep_learning.SiLU(x)[source]

graphax.examples.deep_learning.attn(q, k, v)[source]

graphax.examples.deep_learning.decoder_block(x, q, k, WQ1, WK1, WV1, WQ2, WK2, WV2, W, b, gamma0, gamma1, beta0, beta1)[source]

graphax.examples.deep_learning.encoder_block(x, WQ, WK, WV, W, b, gamma, beta)[source]

graphax.examples.deep_learning.layer_norm(x, gamma, beta)[source]

graphax.examples.deep_learning.sigmoid(x)[source]

graphax.examples.deep_learning.softmax_cross_entropy(logits, labels)[source]

graphax.examples.deep_learning.variance(x, mu, axis=0)[source]

graphax.examples.differential_kinematics module

graphax.examples.differential_kinematics.C(a, b)

graphax.examples.differential_kinematics.RobotArm_6DOF(t1, t2, t3, t4, t5, t6)[source]

graphax.examples.differential_kinematics.S(a, b)

graphax.examples.easy module

graphax.examples.easy.CloudSchemes_step(a1, a2, e1, e2, delta, gamma, bc, br, d1, d2, chi)[source]

graphax.examples.easy.FreeEnergy(x)[source]

graphax.examples.easy.Helmholtz(x)[source]

graphax.examples.easy.Hole(x, y, z, w)[source]

graphax.examples.easy.KerrSenn_Jacobian(t, r, theta, phi)[source]

graphax.examples.easy.KerrSenn_metric(t, r, theta, phi)[source]

graphax.examples.easy.Lighthouse(nu, gamma, omega, t)[source]

graphax.examples.easy.Simple(x, y)[source]

graphax.examples.easy.accretion(a2, bc, br, qc, qr)[source]

graphax.examples.easy.autoconversion(a1, gamma, qc)[source]

graphax.examples.easy.condensation(qc)[source]

graphax.examples.easy.evaporation(e1, d1, e2, d2, qr)[source]

graphax.examples.economics module

graphax.examples.economics.Black76(F, K, r, sigma, T)[source]

graphax.examples.economics.BlackScholes(S, K, r, sigma, T)[source]

graphax.examples.economics.BlackScholes_Jacobian(S, K, r, sigma, T)[source]

graphax.examples.economics.Phi(z)[source]

graphax.examples.minpack module

graphax.examples.minpack.GaussianDataFitting(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11)[source]

graphax.examples.minpack.HumanHeartDipole(x1, x2, x3, x4, x5, x6, x7, x8)[source]

graphax.examples.minpack.PropaneCombustion(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11)[source]

graphax.examples.minpack.SteadyStateCombustion(v00, v01, v10, v11)[source]

graphax.examples.minpack.fL(vij, vip1j, vijp1)[source]

graphax.examples.minpack.fR(vij, vim1j, vijm1)[source]

graphax.examples.minpack.fi(x1, x2, x3, x4, x5, x6, x7, x8, x9, x10, x11, yi, ti)[source]

graphax.examples.minpack.muL(vij, vip1j, vijp1)[source]

graphax.examples.minpack.muR(vij, vim1j, vijm1)[source]

graphax.examples.neuromorphic module

graphax.examples.neuromorphic.ADALIF_SNN(S_in, S_target, U1, U2, U3, a1, a2, a3, W1, W2, W3, alpha, beta, rho, thresh)[source]

graphax.examples.neuromorphic.LIF_SNN(S_in, S_target, U1, U2, U3, I1, I2, I3, W1, W2, W3, alpha, beta, thresh)[source]

graphax.examples.neuromorphic.ada_lif(U, a, S, alpha, beta, rho, threshold)[source]

graphax.examples.neuromorphic.lif(U, I, S, a, b, threshold)[source]

graphax.examples.neuromorphic.superspike_surrogate(beta=10.0)[source]

graphax.examples.randoms module

graphax.examples.randoms.f(v0, v1, v2, v4)[source]

graphax.examples.randoms.g(v0, v1, v2, v3, v4, v5, v6, v7, v8, v9, v10, v11, v12, v13, v14)[source]

graphax.examples.roe module

graphax.examples.roe.F_1d(u0, u1, u2, p)[source]

graphax.examples.roe.F_3d(u0, u, v, u4, p)[source]

graphax.examples.roe.RoeFlux_1d(ul0, ul1, ul2, ur0, ur1, ur2)[source]

graphax.examples.roe.RoeFlux_3d(ul0, ul, ul4, ur0, ur, ur4)[source]

graphax.examples.roe.enthalpy_1d(u0, u4, p)

graphax.examples.roe.enthalpy_3d(u0, u4, p)

graphax.examples.roe.pressure_1d(u0, u1, u2)

graphax.examples.roe.pressure_3d(u0, u, u4)

Module contents