jax.core.ClosedJaxpr

Warning

This page was created from a pull request (#9655).

jax.core.ClosedJaxpr¶

class jax.core.ClosedJaxpr(jaxpr, consts)[source]¶

Parameters

jaxpr (Jaxpr) –
consts (Sequence) –

__init__(jaxpr, consts)[source]¶

Parameters

jaxpr (Jaxpr) –
consts (Sequence) –

Methods

__init__(jaxpr, consts)

param jaxpr

map_jaxpr(f)

pretty_print(*[, source_info, print_shapes])

Attributes

`eqns`
`in_avals`
`literals`
`out_avals`
`jaxpr`
`consts`

previous

jax.core.Jaxpr

next

Autodidax: JAX core from scratch