Package: neptune
neptune.aiQuickstart•Website•Docs•Examples•Resource center•BlogWhat is neptune.ai?Neptune is a lightweight experiment tracker for ML teams that struggle with debugging and reproducing experiments, sharing results, and messy model handover.It offers a single place to track, compare, store, and collaborate on experiments and models.With Neptune, Data Scientists can develop production-ready models faster, and ML Engineers can access model artifacts instantly in order to deploy them to production.Watch a 3min explainer video →Watch a 20min product demo →Play with a live example project in the Neptune app  →Getting startedStep 1:Create afree accountStep 2:Install the Neptune client librarypipinstallneptuneStep 3:Add an experiment tracking snippet to your codeimportneptunerun=neptune.init_run(project="workspace-name/project-name")run["parameters"]={"lr":0.1,"dropout":0.4}run["test_accuracy"]=0.84Core featuresLog and displayAdd a snippet to any step of your ML pipeline once. Decide what and how you want to log. Run a million times.Any framework:any code, fastai, PyTorch, Lightning, TensorFlow/Keras, scikit-learn, 🤗 Transformers, XGBoost, Optuna.Any metadata type:metrics, parameters, dataset and model versions, images, interactive plots, videos, hardware (GPU, CPU, memory), code state.From anywhere in your ML pipeline:multinode pipelines, distributed computing, log during or after execution, log offline, and sync when you are back online.Organize experimentsOrganize logs in a fully customizable nested structure. Display model metadata in user-defined dashboard templates.Nested metadata structure:the flexible API lets you customize the metadata logging structure however you want. Organize nested parameter configs or the results on k-fold validation splits the way they should be.Custom dashboards:combine different metadata types in one view. Define it for one run. Use anywhere. Look at GPU, memory consumption, and load times to debug training speed. See learning curves, image predictions, and confusion matrix to debug model quality.Table views:create different views of the runs table and save them for later. You can have separate table views for debugging, comparing parameter sets, or best experiments.Compare resultsVisualize training live in the neptune.ai web app. See how different parameters and configs affect the results. Optimize models quicker.Compare:learning curves, parameters, images, datasets.Search, sort, and filter:experiments by any field you logged. Use our query language to filter runs based on parameter values, metrics, execution times, or anything else.Visualize and display:runs table, interactive display, folder structure, dashboards.Monitor live:hardware consumption metrics, GPU, CPU, memory.Group by:dataset versions, parameters.Version modelsVersion, review, and access production-ready models and metadata associated with them in a single place.Version models:register models, create model versions, version external model artifacts.Review and change stages:look at the validation, test metrics and other model metadata. You can move models between None/Staging/Production/Archived.Access and share models:every model and model version is accessible via the neptune.ai web app or through the API.Share resultsHave a single place where your team can see the results and access all models and experiments.Send a link:share every chart, dashboard, table view, or anything else you see in the neptune.ai app by copying and sending persistent URLs.Query API:access all model metadata via neptune.ai API. Whatever you logged, you can query in a similar way.Manage users and projects:create different projects, add users to them, and grant different permissions levels.Add your entire org:you can collaborate with a team on every plan, even the Free one. So, invite your entire organization, including product managers and subject matter experts, to increase the visibility from the very beginning.Integrate with any MLOps stackneptune.ai integrates with25+ frameworks:PyTorch, Lightning, TensorFlow/Keras, LightGBM, scikit-learn, XGBoost, Optuna, Kedro, 🤗 Transformers, fastai, Prophet, detectron2, Airflow, and more.PyTorch LightningExample:frompytorch_lightningimportTrainerfromlightning.pytorch.loggersimportNeptuneLogger# Create NeptuneLogger instancefromneptuneimportANONYMOUS_API_TOKENneptune_logger=NeptuneLogger(api_key=ANONYMOUS_API_TOKEN,project="common/pytorch-lightning-integration",tags=["training","resnet"],# optional)# Pass the logger to the Trainertrainer=Trainer(max_epochs=10,logger=neptune_logger)# Run the Trainertrainer.fit(my_model,my_dataloader)neptune.ai is trusted by great companiesRead how various customers use Neptune toimprove their workflow.SupportIf you get stuck or simply want to talk to us about something, here are your options:Check ourFAQ page.Take a look at ourresource center.Chat! In the app, click theblue message iconin the bottom-right corner and send a message. A real person will talk to you ASAP (typically very ASAP).You can just shoot us an email atsupport@neptune.ai.People behindCreated with :heart: by theneptune.ai team