warp-lang 1.10.0__py3-none-macosx_11_0_arm64.whl

warp/fem/__init__.py

@@ -0,0 +1,173 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from . import adaptivity as adaptivity
+from . import cache as cache
+from . import field as field
+from . import geometry as geometry
+from . import linalg as linalg
+from . import operator as operator
+from . import space as space
+from . import utils as utils
+from . import types as types
+
+from warp._src.fem.adaptivity import adaptive_nanogrid_from_field as adaptive_nanogrid_from_field
+from warp._src.fem.adaptivity import adaptive_nanogrid_from_hierarchy as adaptive_nanogrid_from_hierarchy
+
+from warp._src.fem.cache import Temporary as Temporary
+from warp._src.fem.cache import TemporaryStore as TemporaryStore
+from warp._src.fem.cache import borrow_temporary as borrow_temporary
+from warp._src.fem.cache import borrow_temporary_like as borrow_temporary_like
+from warp._src.fem.cache import set_default_temporary_store as set_default_temporary_store
+
+from warp._src.fem.dirichlet import normalize_dirichlet_projector as normalize_dirichlet_projector
+from warp._src.fem.dirichlet import project_linear_system as project_linear_system
+from warp._src.fem.dirichlet import project_system_matrix as project_system_matrix
+from warp._src.fem.dirichlet import project_system_rhs as project_system_rhs
+
+from warp._src.fem.domain import BoundarySides as BoundarySides
+from warp._src.fem.domain import Cells as Cells
+from warp._src.fem.domain import FrontierSides as FrontierSides
+from warp._src.fem.domain import GeometryDomain as GeometryDomain
+from warp._src.fem.domain import Sides as Sides
+from warp._src.fem.domain import Subdomain as Subdomain
+
+from warp._src.fem.field.field import DiscreteField as DiscreteField
+from warp._src.fem.field.field import GeometryField as GeometryField
+from warp._src.fem.field.field import FieldLike as FieldLike
+from warp._src.fem.field.field import ImplicitField as ImplicitField
+from warp._src.fem.field.field import NonconformingField as NonconformingField
+from warp._src.fem.field.field import UniformField as UniformField
+
+from warp._src.fem.field import make_discrete_field as make_discrete_field
+from warp._src.fem.field import make_restriction as make_restriction
+from warp._src.fem.field import make_test as make_test
+from warp._src.fem.field import make_trial as make_trial
+
+from warp._src.fem.geometry.adaptive_nanogrid import AdaptiveNanogrid as AdaptiveNanogrid
+
+from warp._src.fem.geometry.element import Element as Element
+
+from warp._src.fem.geometry.geometry import Geometry as Geometry
+
+from warp._src.fem.geometry.partition import ExplicitGeometryPartition as ExplicitGeometryPartition
+from warp._src.fem.geometry.partition import GeometryPartition as GeometryPartition
+from warp._src.fem.geometry.partition import LinearGeometryPartition as LinearGeometryPartition
+
+from warp._src.fem.geometry.grid_2d import Grid2D as Grid2D
+
+from warp._src.fem.geometry.grid_3d import Grid3D as Grid3D
+
+from warp._src.fem.geometry.hexmesh import Hexmesh as Hexmesh
+
+from warp._src.fem.geometry.nanogrid import Nanogrid as Nanogrid
+
+from warp._src.fem.geometry.quadmesh import Quadmesh2D as Quadmesh2D
+from warp._src.fem.geometry.quadmesh import Quadmesh3D as Quadmesh3D
+
+from warp._src.fem.geometry.tetmesh import Tetmesh as Tetmesh
+
+from warp._src.fem.geometry.trimesh import Trimesh2D as Trimesh2D
+from warp._src.fem.geometry.trimesh import Trimesh3D as Trimesh3D
+
+from warp._src.fem.integrate import integrate as integrate
+from warp._src.fem.integrate import interpolate as interpolate
+
+from warp._src.fem.operator import Integrand as Integrand
+
+from warp._src.fem.operator import D as D
+from warp._src.fem.operator import at_node as at_node
+from warp._src.fem.operator import average as average
+from warp._src.fem.operator import cells as cells
+from warp._src.fem.operator import curl as curl
+from warp._src.fem.operator import deformation_gradient as deformation_gradient
+from warp._src.fem.operator import degree as degree
+from warp._src.fem.operator import div as div
+from warp._src.fem.operator import div_outer as div_outer
+from warp._src.fem.operator import element_closest_point as element_closest_point
+from warp._src.fem.operator import element_coordinates as element_coordinates
+from warp._src.fem.operator import grad as grad
+from warp._src.fem.operator import grad_average as grad_average
+from warp._src.fem.operator import grad_jump as grad_jump
+from warp._src.fem.operator import grad_outer as grad_outer
+from warp._src.fem.operator import inner as inner
+from warp._src.fem.operator import integrand as integrand
+from warp._src.fem.operator import jump as jump
+from warp._src.fem.operator import lookup as lookup
+from warp._src.fem.operator import measure as measure
+from warp._src.fem.operator import measure_ratio as measure_ratio
+from warp._src.fem.operator import node_count as node_count
+from warp._src.fem.operator import node_index as node_index
+from warp._src.fem.operator import normal as normal
+from warp._src.fem.operator import outer as outer
+from warp._src.fem.operator import partition_lookup as partition_lookup
+from warp._src.fem.operator import position as position
+from warp._src.fem.operator import to_cell_side as to_cell_side
+from warp._src.fem.operator import to_inner_cell as to_inner_cell
+from warp._src.fem.operator import to_outer_cell as to_outer_cell
+
+from warp._src.fem.polynomial import Polynomial as Polynomial
+
+from warp._src.fem.quadrature.pic_quadrature import PicQuadrature as PicQuadrature
+
+from warp._src.fem.quadrature.quadrature import ExplicitQuadrature as ExplicitQuadrature
+from warp._src.fem.quadrature.quadrature import NodalQuadrature as NodalQuadrature
+from warp._src.fem.quadrature.quadrature import Quadrature as Quadrature
+from warp._src.fem.quadrature.quadrature import RegularQuadrature as RegularQuadrature
+
+from warp._src.fem.space.basis_space import BasisSpace as BasisSpace
+from warp._src.fem.space.basis_space import PointBasisSpace as PointBasisSpace
+from warp._src.fem.space.basis_space import ShapeBasisSpace as ShapeBasisSpace
+
+from warp._src.fem.space.dof_mapper import DofMapper as DofMapper
+from warp._src.fem.space.dof_mapper import SkewSymmetricTensorMapper as SkewSymmetricTensorMapper
+from warp._src.fem.space.dof_mapper import SymmetricTensorMapper as SymmetricTensorMapper
+
+from warp._src.fem.space.function_space import FunctionSpace as FunctionSpace
+
+from warp._src.fem.space.partition import SpacePartition as SpacePartition
+from warp._src.fem.space.partition import make_space_partition as make_space_partition
+
+from warp._src.fem.space.restriction import SpaceRestriction as SpaceRestriction
+
+from warp._src.fem.space.shape import ElementBasis as ElementBasis
+from warp._src.fem.space.shape import ShapeFunction as ShapeFunction
+
+from warp._src.fem.space import make_collocated_function_space as make_collocated_function_space
+from warp._src.fem.space import make_contravariant_function_space as make_contravariant_function_space
+from warp._src.fem.space import make_covariant_function_space as make_covariant_function_space
+from warp._src.fem.space import make_element_shape_function as make_element_shape_function
+from warp._src.fem.space import make_element_based_space_topology as make_element_based_space_topology
+from warp._src.fem.space import make_polynomial_basis_space as make_polynomial_basis_space
+from warp._src.fem.space import make_polynomial_space as make_polynomial_space
+from warp._src.fem.space import make_space_restriction as make_space_restriction
+
+from warp._src.fem.space.topology import SpaceTopology as SpaceTopology
+
+from warp._src.fem.types import NULL_ELEMENT_INDEX as NULL_ELEMENT_INDEX
+from warp._src.fem.types import NULL_NODE_INDEX as NULL_NODE_INDEX
+from warp._src.fem.types import NULL_QP_INDEX as NULL_QP_INDEX
+from warp._src.fem.types import OUTSIDE as OUTSIDE
+from warp._src.fem.types import Coords as Coords
+from warp._src.fem.types import Domain as Domain
+from warp._src.fem.types import ElementIndex as ElementIndex
+from warp._src.fem.types import ElementKind as ElementKind
+from warp._src.fem.types import Field as Field
+from warp._src.fem.types import QuadraturePointIndex as QuadraturePointIndex
+from warp._src.fem.types import NodeIndex as NodeIndex
+from warp._src.fem.types import Sample as Sample
+from warp._src.fem.types import make_free_sample as make_free_sample

warp/fem/adaptivity.py

@@ -0,0 +1,26 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import adaptivity as _adaptivity
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_adaptivity, "wp.fem", name)

warp/fem/cache.py

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.cache import dynamic_kernel as dynamic_kernel
+from warp._src.fem.cache import dynamic_func as dynamic_func
+
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import cache as _cache
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_cache, "wp.fem", name)

warp/fem/dirichlet.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import dirichlet as _dirichlet
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_dirichlet, "wp.fem", name)

warp/fem/field/__init__.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from . import field as field
+
+from warp._src.fem.field import FieldRestriction as FieldRestriction
+from warp._src.fem.field import GeometryField as GeometryField
+from warp._src.fem.field import SpaceField as SpaceField
+from warp._src.fem.field import TestField as TestField
+from warp._src.fem.field import TrialField as TrialField

warp/fem/field/field.py

@@ -0,0 +1,26 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem.field import field as _field
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_field, "wp.fem.field", name)

warp/fem/geometry/__init__.py

@@ -0,0 +1,21 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from . import closest_point as closest_point
+
+from warp._src.fem.geometry import DeformedGeometry as DeformedGeometry
+from warp._src.fem.geometry import WholeGeometryPartition as WholeGeometryPartition

warp/fem/geometry/closest_point.py

@@ -0,0 +1,31 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.geometry.closest_point import project_on_seg_at_origin as project_on_seg_at_origin
+from warp._src.fem.geometry.closest_point import project_on_tri_at_origin as project_on_tri_at_origin
+from warp._src.fem.geometry.closest_point import project_on_tet_at_origin as project_on_tet_at_origin
+from warp._src.fem.geometry.closest_point import project_on_box_at_origin as project_on_box_at_origin
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem.geometry import closest_point as _closest_point
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_closest_point, "wp.fem.geometry", name)

warp/fem/linalg.py

@@ -0,0 +1,38 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.linalg import array_axpy as array_axpy
+from warp._src.fem.linalg import generalized_inner as generalized_inner
+from warp._src.fem.linalg import generalized_outer as generalized_outer
+from warp._src.fem.linalg import spherical_part as spherical_part
+from warp._src.fem.linalg import symmetric_part as symmetric_part
+from warp._src.fem.linalg import skew_part as skew_part
+from warp._src.fem.linalg import householder_make_hessenberg as householder_make_hessenberg
+from warp._src.fem.linalg import householder_qr_decomposition as householder_qr_decomposition
+from warp._src.fem.linalg import inverse_qr as inverse_qr
+from warp._src.fem.linalg import solve_triangular as solve_triangular
+from warp._src.fem.linalg import symmetric_eigenvalues_qr as symmetric_eigenvalues_qr
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import linalg as _linalg
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_linalg, "wp.fem", name)

warp/fem/operator.py

@@ -0,0 +1,32 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.operator import element_partition_index as element_partition_index
+from warp._src.fem.operator import node_partition_index as node_partition_index
+from warp._src.fem.operator import operator as operator
+from warp._src.fem.operator import Operator as Operator
+
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import operator as _operator
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_operator, "wp.fem", name)

warp/fem/polynomial.py

@@ -0,0 +1,29 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.polynomial import quadrature_1d as quadrature_1d
+from warp._src.fem.polynomial import lagrange_scales as lagrange_scales
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import polynomial as _polynomial
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_polynomial, "wp.fem", name)

warp/fem/space/__init__.py

@@ -0,0 +1,22 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.space import CollocatedFunctionSpace as CollocatedFunctionSpace
+from warp._src.fem.space import ContravariantFunctionSpace as ContravariantFunctionSpace
+from warp._src.fem.space import CovariantFunctionSpace as CovariantFunctionSpace
+
+from . import shape as shape

warp/fem/space/basis_space.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem.space import basis_space as _basis_space
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_basis_space, "wp.fem.space", name)

warp/fem/space/shape/__init__.py

@@ -0,0 +1,68 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+# Keeping those imports non-deprecated for now,
+# Will be deprecated after unification of the shape functions API (1.11+)
+
+from warp._src.fem.space.shape import CubeShapeFunction as CubeShapeFunction
+from warp._src.fem.space.shape import CubeNedelecFirstKindShapeFunctions as CubeNedelecFirstKindShapeFunctions
+from warp._src.fem.space.shape import (
+    CubeNonConformingPolynomialShapeFunctions as CubeNonConformingPolynomialShapeFunctions,
+)
+from warp._src.fem.space.shape import CubeRaviartThomasShapeFunctions as CubeRaviartThomasShapeFunctions
+from warp._src.fem.space.shape import CubeSerendipityShapeFunctions as CubeSerendipityShapeFunctions
+from warp._src.fem.space.shape import CubeTripolynomialShapeFunctions as CubeTripolynomialShapeFunctions
+
+from warp._src.fem.space.shape import SquareShapeFunction as SquareShapeFunction
+from warp._src.fem.space.shape import SquareBipolynomialShapeFunctions as SquareBipolynomialShapeFunctions
+from warp._src.fem.space.shape import SquareNedelecFirstKindShapeFunctions as SquareNedelecFirstKindShapeFunctions
+from warp._src.fem.space.shape import (
+    SquareNonConformingPolynomialShapeFunctions as SquareNonConformingPolynomialShapeFunctions,
+)
+from warp._src.fem.space.shape import SquareRaviartThomasShapeFunctions as SquareRaviartThomasShapeFunctions
+from warp._src.fem.space.shape import SquareSerendipityShapeFunctions as SquareSerendipityShapeFunctions
+
+from warp._src.fem.space.shape import TriangleShapeFunction as TriangleShapeFunction
+from warp._src.fem.space.shape import TriangleNedelecFirstKindShapeFunctions as TriangleNedelecFirstKindShapeFunctions
+from warp._src.fem.space.shape import (
+    TriangleNonConformingPolynomialShapeFunctions as TriangleNonConformingPolynomialShapeFunctions,
+)
+from warp._src.fem.space.shape import TrianglePolynomialShapeFunctions as TrianglePolynomialShapeFunctions
+from warp._src.fem.space.shape import TriangleRaviartThomasShapeFunctions as TriangleRaviartThomasShapeFunctions
+
+from warp._src.fem.space.shape import TetrahedronShapeFunction as TetrahedronShapeFunction
+from warp._src.fem.space.shape import (
+    TetrahedronNedelecFirstKindShapeFunctions as TetrahedronNedelecFirstKindShapeFunctions,
+)
+from warp._src.fem.space.shape import (
+    TetrahedronNonConformingPolynomialShapeFunctions as TetrahedronNonConformingPolynomialShapeFunctions,
+)
+from warp._src.fem.space.shape import TetrahedronPolynomialShapeFunctions as TetrahedronPolynomialShapeFunctions
+from warp._src.fem.space.shape import TetrahedronRaviartThomasShapeFunctions as TetrahedronRaviartThomasShapeFunctions
+
+from warp._src.fem.space.shape import ConstantShapeFunction as ConstantShapeFunction
+
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem.space import shape as _shape
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_shape, "wp.fem.space", name)

warp/fem/space/topology.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem.space import topology as _topology
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_topology, "wp.fem.space", name)

warp/fem/types.py

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import types as _types
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_types, "wp.fem", name)

warp/fem/utils.py

@@ -0,0 +1,32 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# isort: skip_file
+
+from warp._src.fem.utils import grid_to_quads as grid_to_quads
+from warp._src.fem.utils import grid_to_tris as grid_to_tris
+from warp._src.fem.utils import grid_to_tets as grid_to_tets
+from warp._src.fem.utils import grid_to_hexes as grid_to_hexes
+
+
+# TODO: Remove after cleaning up the public API.
+
+from warp._src.fem import utils as _utils
+
+
+def __getattr__(name):
+    from warp._src.utils import get_deprecated_api
+
+    return get_deprecated_api(_utils, "wp.fem", name)