import torch import torch.nn as nn import torch.optim as optim import numpy as np import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import plotly.graph_objects as go from plotly.subplots import make_subplots from tqdm import tqdm import warnings warnings...