├── .devcontainer ├── Dockerfile ├── base.Dockerfile ├── devcontainer.json └── library-scripts │ ├── README.md │ ├── common-debian.sh │ ├── gradle-debian.sh │ ├── java-debian.sh │ ├── maven-debian.sh │ └── node-debian.sh ├── .github ├── CODEOWNERS ├── ISSUE_TEMPLATE │ ├── bug_report.md │ └── feature_request.md ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── codeql-analysis-java.yml │ ├── continuous.yml │ ├── docker_publish.yml │ ├── docs.yml │ ├── native_build_mxnet_osx.yml │ ├── native_jni_s3_huggingface_android.yml │ ├── native_jni_s3_pytorch.yml │ ├── native_jni_s3_pytorch_android.yml │ ├── native_jni_s3_tensorrt.yml │ ├── native_publish_mxnet.yml │ ├── native_publish_pytorch.yml │ ├── native_publish_tensorflow.yml │ ├── native_s3_fasttext.yml │ ├── native_s3_huggingface.yml │ ├── native_s3_pytorch.yml │ ├── native_s3_pytorch_android.yml │ ├── native_s3_sentencepiece.yml │ ├── native_s3_tensorflow.yml │ ├── native_s3_xgboost.yml │ ├── nightly_android.yml │ ├── nightly_publish.yml │ ├── publish-job-success.yml │ ├── publish_android_packages.yml │ └── serving-publish.yml ├── .gitignore ├── CNAME ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README.md ├── android ├── .gitignore ├── README.md ├── build.gradle ├── core │ ├── build.gradle │ ├── consumer-rules.pro │ ├── gradlew │ └── src │ │ ├── androidTest │ │ └── java │ │ │ └── ai │ │ │ └── djl │ │ │ └── android │ │ │ └── core │ │ │ ├── BitmapWrapperTest.java │ │ │ └── ModelLoadingTest.java │ │ └── main │ │ ├── AndroidManifest.xml │ │ └── java │ │ └── ai │ │ └── djl │ │ └── android │ │ └── core │ │ └── BitmapImageFactory.java ├── gradle ├── gradle.properties ├── gradlew ├── gradlew.bat ├── pytorch-native │ ├── README.md │ ├── build.gradle │ ├── consumer-rules.pro │ ├── gradlew │ ├── proguard-rules.pro │ └── src │ │ └── main │ │ └── AndroidManifest.xml ├── settings.gradle └── tokenizer-native │ ├── README.md │ ├── build.gradle │ ├── consumer-rules.pro │ ├── gradlew │ ├── proguard-rules.pro │ └── src │ └── main │ └── AndroidManifest.xml ├── api ├── README.md ├── build.gradle.kts ├── gradlew └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ ├── Application.java │ │ │ ├── BaseModel.java │ │ │ ├── Device.java │ │ │ ├── MalformedModelException.java │ │ │ ├── Model.java │ │ │ ├── ModelException.java │ │ │ ├── TrainingDivergedException.java │ │ │ ├── engine │ │ │ ├── Engine.java │ │ │ ├── EngineException.java │ │ │ ├── EngineProvider.java │ │ │ ├── StandardCapabilities.java │ │ │ └── package-info.java │ │ │ ├── inference │ │ │ ├── Predictor.java │ │ │ ├── package-info.java │ │ │ └── streaming │ │ │ │ ├── ChunkedBytesSupplier.java │ │ │ │ ├── IteratorBytesSupplier.java │ │ │ │ ├── PublisherBytesSupplier.java │ │ │ │ ├── StreamingBlock.java │ │ │ │ ├── StreamingTranslator.java │ │ │ │ └── package-info.java │ │ │ ├── metric │ │ │ ├── Dimension.java │ │ │ ├── Metric.java │ │ │ ├── MetricType.java │ │ │ ├── Metrics.java │ │ │ ├── Unit.java │ │ │ └── package-info.java │ │ │ ├── modality │ │ │ ├── Classifications.java │ │ │ ├── Input.java │ │ │ ├── Output.java │ │ │ ├── audio │ │ │ │ ├── Audio.java │ │ │ │ ├── AudioFactory.java │ │ │ │ ├── SampledAudioFactory.java │ │ │ │ ├── package-info.java │ │ │ │ └── translator │ │ │ │ │ ├── SpeechRecognitionTranslator.java │ │ │ │ │ ├── SpeechRecognitionTranslatorFactory.java │ │ │ │ │ └── package-info.java │ │ │ ├── cv │ │ │ │ ├── BufferedImageFactory.java │ │ │ │ ├── Image.java │ │ │ │ ├── ImageFactory.java │ │ │ │ ├── MultiBoxDetection.java │ │ │ │ ├── MultiBoxPrior.java │ │ │ │ ├── MultiBoxTarget.java │ │ │ │ ├── VisionLanguageInput.java │ │ │ │ ├── output │ │ │ │ │ ├── BoundingBox.java │ │ │ │ │ ├── CategoryMask.java │ │ │ │ │ ├── DetectedObjects.java │ │ │ │ │ ├── Joints.java │ │ │ │ │ ├── Landmark.java │ │ │ │ │ ├── Mask.java │ │ │ │ │ ├── Point.java │ │ │ │ │ ├── Rectangle.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ ├── transform │ │ │ │ │ ├── CenterCrop.java │ │ │ │ │ ├── CenterFit.java │ │ │ │ │ ├── Crop.java │ │ │ │ │ ├── Normalize.java │ │ │ │ │ ├── OneHot.java │ │ │ │ │ ├── Pad.java │ │ │ │ │ ├── RandomBrightness.java │ │ │ │ │ ├── RandomColorJitter.java │ │ │ │ │ ├── RandomFlipLeftRight.java │ │ │ │ │ ├── RandomFlipTopBottom.java │ │ │ │ │ ├── RandomHue.java │ │ │ │ │ ├── RandomResizedCrop.java │ │ │ │ │ ├── Resize.java │ │ │ │ │ ├── ResizeShort.java │ │ │ │ │ ├── ToTensor.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── translator │ │ │ │ │ ├── BaseImagePreProcessor.java │ │ │ │ │ ├── BaseImageTranslator.java │ │ │ │ │ ├── BaseImageTranslatorFactory.java │ │ │ │ │ ├── BigGANTranslator.java │ │ │ │ │ ├── BigGANTranslatorFactory.java │ │ │ │ │ ├── ImageClassificationTranslator.java │ │ │ │ │ ├── ImageClassificationTranslatorFactory.java │ │ │ │ │ ├── ImageFeatureExtractor.java │ │ │ │ │ ├── ImageFeatureExtractorFactory.java │ │ │ │ │ ├── ImageServingTranslator.java │ │ │ │ │ ├── InstanceSegmentationTranslator.java │ │ │ │ │ ├── InstanceSegmentationTranslatorFactory.java │ │ │ │ │ ├── ObjectDetectionTranslator.java │ │ │ │ │ ├── ObjectDetectionTranslatorFactory.java │ │ │ │ │ ├── Sam2ServingTranslator.java │ │ │ │ │ ├── Sam2Translator.java │ │ │ │ │ ├── Sam2TranslatorFactory.java │ │ │ │ │ ├── SemanticSegmentationTranslator.java │ │ │ │ │ ├── SemanticSegmentationTranslatorFactory.java │ │ │ │ │ ├── SimplePoseTranslator.java │ │ │ │ │ ├── SimplePoseTranslatorFactory.java │ │ │ │ │ ├── SingleShotDetectionTranslator.java │ │ │ │ │ ├── SingleShotDetectionTranslatorFactory.java │ │ │ │ │ ├── StyleTransferTranslator.java │ │ │ │ │ ├── StyleTransferTranslatorFactory.java │ │ │ │ │ ├── YoloPoseTranslator.java │ │ │ │ │ ├── YoloPoseTranslatorFactory.java │ │ │ │ │ ├── YoloSegmentationTranslator.java │ │ │ │ │ ├── YoloSegmentationTranslatorFactory.java │ │ │ │ │ ├── YoloTranslator.java │ │ │ │ │ ├── YoloTranslatorFactory.java │ │ │ │ │ ├── YoloV5Translator.java │ │ │ │ │ ├── YoloV5TranslatorFactory.java │ │ │ │ │ ├── YoloV8Translator.java │ │ │ │ │ ├── YoloV8TranslatorFactory.java │ │ │ │ │ ├── YoloWorldTranslator.java │ │ │ │ │ ├── YoloWorldTranslatorFactory.java │ │ │ │ │ ├── ZeroShotImageClassificationServingTranslator.java │ │ │ │ │ ├── ZeroShotObjectDetectionServingTranslator.java │ │ │ │ │ ├── package-info.java │ │ │ │ │ └── wrapper │ │ │ │ │ │ ├── FileImagePreProcessor.java │ │ │ │ │ │ ├── InputStreamImagePreProcessor.java │ │ │ │ │ │ ├── StringImagePreProcessor.java │ │ │ │ │ │ ├── UrlImagePreProcessor.java │ │ │ │ │ │ └── package-info.java │ │ │ │ └── util │ │ │ │ │ ├── NDImageUtils.java │ │ │ │ │ └── package-info.java │ │ │ ├── nlp │ │ │ │ ├── Decoder.java │ │ │ │ ├── DefaultVocabulary.java │ │ │ │ ├── EmbeddingOutput.java │ │ │ │ ├── Encoder.java │ │ │ │ ├── EncoderDecoder.java │ │ │ │ ├── NlpUtils.java │ │ │ │ ├── TextPrompt.java │ │ │ │ ├── Vocabulary.java │ │ │ │ ├── bert │ │ │ │ │ ├── BertFullTokenizer.java │ │ │ │ │ ├── BertToken.java │ │ │ │ │ ├── BertTokenizer.java │ │ │ │ │ ├── WordpieceTokenizer.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── embedding │ │ │ │ │ ├── EmbeddingException.java │ │ │ │ │ ├── ModelZooTextEmbedding.java │ │ │ │ │ ├── SimpleTextEmbedding.java │ │ │ │ │ ├── TextEmbedding.java │ │ │ │ │ ├── TrainableTextEmbedding.java │ │ │ │ │ ├── TrainableWordEmbedding.java │ │ │ │ │ ├── WordEmbedding.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── generate │ │ │ │ │ ├── BatchTensorList.java │ │ │ │ │ ├── BeamBatchTensorList.java │ │ │ │ │ ├── CausalLMOutput.java │ │ │ │ │ ├── ContrastiveBatchTensorList.java │ │ │ │ │ ├── ContrastiveSeqBatchScheduler.java │ │ │ │ │ ├── GreedyBatchTensorList.java │ │ │ │ │ ├── SearchConfig.java │ │ │ │ │ ├── SeqBatchScheduler.java │ │ │ │ │ ├── SeqBatcher.java │ │ │ │ │ ├── StepGeneration.java │ │ │ │ │ ├── TextGenerator.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ ├── preprocess │ │ │ │ │ ├── HyphenNormalizer.java │ │ │ │ │ ├── LambdaProcessor.java │ │ │ │ │ ├── LowerCaseConvertor.java │ │ │ │ │ ├── PunctuationSeparator.java │ │ │ │ │ ├── SimpleTokenizer.java │ │ │ │ │ ├── TextCleaner.java │ │ │ │ │ ├── TextProcessor.java │ │ │ │ │ ├── TextTerminator.java │ │ │ │ │ ├── TextTruncator.java │ │ │ │ │ ├── Tokenizer.java │ │ │ │ │ ├── UnicodeNormalizer.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── qa │ │ │ │ │ ├── QAInput.java │ │ │ │ │ └── package-info.java │ │ │ │ └── translator │ │ │ │ │ ├── CrossEncoderServingTranslator.java │ │ │ │ │ ├── NamedEntity.java │ │ │ │ │ ├── QATranslator.java │ │ │ │ │ ├── QaServingTranslator.java │ │ │ │ │ ├── SimpleText2TextTranslator.java │ │ │ │ │ ├── SparseRetrievalServingTranslator.java │ │ │ │ │ ├── TextClassificationServingTranslator.java │ │ │ │ │ ├── TextEmbeddingServingTranslator.java │ │ │ │ │ ├── TokenClassificationServingTranslator.java │ │ │ │ │ ├── ZeroShotClassificationInput.java │ │ │ │ │ ├── ZeroShotClassificationOutput.java │ │ │ │ │ ├── ZeroShotClassificationServingTranslator.java │ │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── rl │ │ │ │ ├── ActionSpace.java │ │ │ │ ├── LruReplayBuffer.java │ │ │ │ ├── ReplayBuffer.java │ │ │ │ ├── agent │ │ │ │ ├── EpsilonGreedy.java │ │ │ │ ├── QAgent.java │ │ │ │ ├── RlAgent.java │ │ │ │ └── package-info.java │ │ │ │ ├── env │ │ │ │ ├── RlEnv.java │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ │ ├── ndarray │ │ │ ├── BaseNDManager.java │ │ │ ├── BytesSupplier.java │ │ │ ├── BytesSupplierImpl.java │ │ │ ├── LazyNDArray.java │ │ │ ├── NDArray.java │ │ │ ├── NDArrayAdapter.java │ │ │ ├── NDArrays.java │ │ │ ├── NDList.java │ │ │ ├── NDManager.java │ │ │ ├── NDResource.java │ │ │ ├── NDScope.java │ │ │ ├── NDSerializer.java │ │ │ ├── NDUtils.java │ │ │ ├── SparseNDArray.java │ │ │ ├── index │ │ │ │ ├── NDArrayIndexer.java │ │ │ │ ├── NDIndex.java │ │ │ │ ├── dim │ │ │ │ │ ├── NDIndexAll.java │ │ │ │ │ ├── NDIndexBooleans.java │ │ │ │ │ ├── NDIndexElement.java │ │ │ │ │ ├── NDIndexFixed.java │ │ │ │ │ ├── NDIndexNull.java │ │ │ │ │ ├── NDIndexPick.java │ │ │ │ │ ├── NDIndexSlice.java │ │ │ │ │ ├── NDIndexTake.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── full │ │ │ │ │ ├── NDIndexFullPick.java │ │ │ │ │ ├── NDIndexFullSlice.java │ │ │ │ │ ├── NDIndexFullTake.java │ │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ │ ├── internal │ │ │ │ ├── NDArrayEx.java │ │ │ │ ├── NDFormat.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── types │ │ │ │ ├── DataDesc.java │ │ │ │ ├── DataType.java │ │ │ │ ├── LayoutType.java │ │ │ │ ├── Shape.java │ │ │ │ ├── SparseFormat.java │ │ │ │ └── package-info.java │ │ │ ├── nn │ │ │ ├── AbstractBaseBlock.java │ │ │ ├── AbstractBlock.java │ │ │ ├── AbstractSymbolBlock.java │ │ │ ├── Activation.java │ │ │ ├── Block.java │ │ │ ├── BlockFactory.java │ │ │ ├── BlockList.java │ │ │ ├── Blocks.java │ │ │ ├── IdentityBlockFactory.java │ │ │ ├── LambdaBlock.java │ │ │ ├── OnesBlockFactory.java │ │ │ ├── ParallelBlock.java │ │ │ ├── Parameter.java │ │ │ ├── ParameterList.java │ │ │ ├── SequentialBlock.java │ │ │ ├── SymbolBlock.java │ │ │ ├── UninitializedParameterException.java │ │ │ ├── convolutional │ │ │ │ ├── Conv1d.java │ │ │ │ ├── Conv1dTranspose.java │ │ │ │ ├── Conv2d.java │ │ │ │ ├── Conv2dTranspose.java │ │ │ │ ├── Conv3d.java │ │ │ │ ├── Convolution.java │ │ │ │ ├── Deconvolution.java │ │ │ │ └── package-info.java │ │ │ ├── core │ │ │ │ ├── AbstractEmbedding.java │ │ │ │ ├── AbstractIndexedEmbedding.java │ │ │ │ ├── ConstantEmbedding.java │ │ │ │ ├── Embedding.java │ │ │ │ ├── Linear.java │ │ │ │ ├── LinearCollection.java │ │ │ │ ├── Multiplication.java │ │ │ │ ├── Prelu.java │ │ │ │ ├── SparseMax.java │ │ │ │ └── package-info.java │ │ │ ├── norm │ │ │ │ ├── BatchNorm.java │ │ │ │ ├── Dropout.java │ │ │ │ ├── GhostBatchNorm.java │ │ │ │ ├── LayerNorm.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── pooling │ │ │ │ ├── Pool.java │ │ │ │ └── package-info.java │ │ │ ├── recurrent │ │ │ │ ├── GRU.java │ │ │ │ ├── LSTM.java │ │ │ │ ├── RNN.java │ │ │ │ ├── RecurrentBlock.java │ │ │ │ └── package-info.java │ │ │ └── transformer │ │ │ │ ├── BertBlock.java │ │ │ │ ├── BertMaskedLanguageModelBlock.java │ │ │ │ ├── BertMaskedLanguageModelLoss.java │ │ │ │ ├── BertNextSentenceBlock.java │ │ │ │ ├── BertNextSentenceLoss.java │ │ │ │ ├── BertPretrainingBlock.java │ │ │ │ ├── BertPretrainingLoss.java │ │ │ │ ├── IdEmbedding.java │ │ │ │ ├── PointwiseFeedForwardBlock.java │ │ │ │ ├── ScaledDotProductAttentionBlock.java │ │ │ │ ├── TransformerEncoderBlock.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── repository │ │ │ ├── AbstractRepository.java │ │ │ ├── Artifact.java │ │ │ ├── FilenameUtils.java │ │ │ ├── JarRepository.java │ │ │ ├── License.java │ │ │ ├── LocalRepository.java │ │ │ ├── MRL.java │ │ │ ├── Metadata.java │ │ │ ├── RemoteRepository.java │ │ │ ├── Repository.java │ │ │ ├── RepositoryFactory.java │ │ │ ├── RepositoryFactoryImpl.java │ │ │ ├── Restriction.java │ │ │ ├── SimpleRepository.java │ │ │ ├── SimpleUrlRepository.java │ │ │ ├── Version.java │ │ │ ├── VersionRange.java │ │ │ ├── package-info.java │ │ │ └── zoo │ │ │ │ ├── BaseModelLoader.java │ │ │ │ ├── Criteria.java │ │ │ │ ├── DefaultModelZoo.java │ │ │ │ ├── DefaultZooProvider.java │ │ │ │ ├── ModelLoader.java │ │ │ │ ├── ModelNotFoundException.java │ │ │ │ ├── ModelZoo.java │ │ │ │ ├── ModelZooResolver.java │ │ │ │ ├── ZooModel.java │ │ │ │ ├── ZooProvider.java │ │ │ │ ├── ZooProviderNotFoundException.java │ │ │ │ └── package-info.java │ │ │ ├── training │ │ │ ├── DefaultTrainingConfig.java │ │ │ ├── EasyTrain.java │ │ │ ├── GradientCollector.java │ │ │ ├── LocalParameterServer.java │ │ │ ├── ParameterServer.java │ │ │ ├── ParameterStore.java │ │ │ ├── Trainer.java │ │ │ ├── TrainingConfig.java │ │ │ ├── TrainingResult.java │ │ │ ├── dataset │ │ │ │ ├── ArrayDataset.java │ │ │ │ ├── Batch.java │ │ │ │ ├── BatchSampler.java │ │ │ │ ├── BulkDataIterable.java │ │ │ │ ├── DataIterable.java │ │ │ │ ├── Dataset.java │ │ │ │ ├── RandomAccessDataset.java │ │ │ │ ├── RandomSampler.java │ │ │ │ ├── RawDataset.java │ │ │ │ ├── Record.java │ │ │ │ ├── Sampler.java │ │ │ │ ├── SequenceSampler.java │ │ │ │ └── package-info.java │ │ │ ├── evaluator │ │ │ │ ├── AbstractAccuracy.java │ │ │ │ ├── Accuracy.java │ │ │ │ ├── BinaryAccuracy.java │ │ │ │ ├── BoundingBoxError.java │ │ │ │ ├── Coverage.java │ │ │ │ ├── Evaluator.java │ │ │ │ ├── IndexEvaluator.java │ │ │ │ ├── SingleShotDetectionAccuracy.java │ │ │ │ ├── TopKAccuracy.java │ │ │ │ └── package-info.java │ │ │ ├── hyperparameter │ │ │ │ ├── EasyHpo.java │ │ │ │ ├── optimizer │ │ │ │ │ ├── BaseHpOptimizer.java │ │ │ │ │ ├── HpORandom.java │ │ │ │ │ ├── HpOptimizer.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ └── param │ │ │ │ │ ├── HpBool.java │ │ │ │ │ ├── HpCategorical.java │ │ │ │ │ ├── HpFloat.java │ │ │ │ │ ├── HpInt.java │ │ │ │ │ ├── HpSet.java │ │ │ │ │ ├── HpVal.java │ │ │ │ │ ├── Hyperparameter.java │ │ │ │ │ └── package-info.java │ │ │ ├── initializer │ │ │ │ ├── ConstantInitializer.java │ │ │ │ ├── Initializer.java │ │ │ │ ├── NormalInitializer.java │ │ │ │ ├── TruncatedNormalInitializer.java │ │ │ │ ├── UniformInitializer.java │ │ │ │ ├── XavierInitializer.java │ │ │ │ └── package-info.java │ │ │ ├── listener │ │ │ │ ├── DivergenceCheckTrainingListener.java │ │ │ │ ├── EarlyStoppingListener.java │ │ │ │ ├── EpochTrainingListener.java │ │ │ │ ├── EvaluatorTrainingListener.java │ │ │ │ ├── LoggingTrainingListener.java │ │ │ │ ├── MemoryTrainingListener.java │ │ │ │ ├── SaveModelTrainingListener.java │ │ │ │ ├── TimeMeasureTrainingListener.java │ │ │ │ ├── TrainingListener.java │ │ │ │ ├── TrainingListenerAdapter.java │ │ │ │ └── package-info.java │ │ │ ├── loss │ │ │ │ ├── AbstractCompositeLoss.java │ │ │ │ ├── ElasticNetWeightDecay.java │ │ │ │ ├── HingeLoss.java │ │ │ │ ├── IndexLoss.java │ │ │ │ ├── L1Loss.java │ │ │ │ ├── L1WeightDecay.java │ │ │ │ ├── L2Loss.java │ │ │ │ ├── L2WeightDecay.java │ │ │ │ ├── Loss.java │ │ │ │ ├── MaskedSoftmaxCrossEntropyLoss.java │ │ │ │ ├── QuantileL1Loss.java │ │ │ │ ├── SigmoidBinaryCrossEntropyLoss.java │ │ │ │ ├── SimpleCompositeLoss.java │ │ │ │ ├── SingleShotDetectionLoss.java │ │ │ │ ├── SoftmaxCrossEntropyLoss.java │ │ │ │ ├── TabNetClassificationLoss.java │ │ │ │ ├── TabNetRegressionLoss.java │ │ │ │ ├── YOLOv3Loss.java │ │ │ │ └── package-info.java │ │ │ ├── optimizer │ │ │ │ ├── Adadelta.java │ │ │ │ ├── Adagrad.java │ │ │ │ ├── Adam.java │ │ │ │ ├── AdamW.java │ │ │ │ ├── Nag.java │ │ │ │ ├── Optimizer.java │ │ │ │ ├── RmsProp.java │ │ │ │ ├── Sgd.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── tracker │ │ │ │ ├── CosineTracker.java │ │ │ │ ├── CyclicalTracker.java │ │ │ │ ├── FactorTracker.java │ │ │ │ ├── FixedPerVarTracker.java │ │ │ │ ├── FixedTracker.java │ │ │ │ ├── LinearTracker.java │ │ │ │ ├── MultiFactorTracker.java │ │ │ │ ├── ParameterTracker.java │ │ │ │ ├── PolynomialDecayTracker.java │ │ │ │ ├── Tracker.java │ │ │ │ ├── WarmUpTracker.java │ │ │ │ └── package-info.java │ │ │ └── util │ │ │ │ ├── DownloadUtils.java │ │ │ │ ├── MinMaxScaler.java │ │ │ │ ├── ProgressBar.java │ │ │ │ └── package-info.java │ │ │ ├── translate │ │ │ ├── ArgumentsUtil.java │ │ │ ├── BasicTranslator.java │ │ │ ├── Batchifier.java │ │ │ ├── DefaultTranslatorFactory.java │ │ │ ├── DeferredTranslatorFactory.java │ │ │ ├── Ensembleable.java │ │ │ ├── ExpansionTranslatorFactory.java │ │ │ ├── NDArraySupplier.java │ │ │ ├── NoBatchifyTranslator.java │ │ │ ├── NoopServingTranslatorFactory.java │ │ │ ├── NoopTranslator.java │ │ │ ├── PaddingStackBatchifier.java │ │ │ ├── Pipeline.java │ │ │ ├── PostProcessor.java │ │ │ ├── PreProcessor.java │ │ │ ├── ServingTranslator.java │ │ │ ├── ServingTranslatorFactory.java │ │ │ ├── SimplePaddingStackBatchifier.java │ │ │ ├── StackBatchifier.java │ │ │ ├── Transform.java │ │ │ ├── TranslateException.java │ │ │ ├── Translator.java │ │ │ ├── TranslatorContext.java │ │ │ ├── TranslatorFactory.java │ │ │ ├── TranslatorOptions.java │ │ │ └── package-info.java │ │ │ └── util │ │ │ ├── ClassLoaderUtils.java │ │ │ ├── Ec2Utils.java │ │ │ ├── Float16Utils.java │ │ │ ├── Hex.java │ │ │ ├── JsonSerializable.java │ │ │ ├── JsonUtils.java │ │ │ ├── NativeResource.java │ │ │ ├── NeuronUtils.java │ │ │ ├── Pair.java │ │ │ ├── PairList.java │ │ │ ├── Platform.java │ │ │ ├── Preconditions.java │ │ │ ├── Progress.java │ │ │ ├── RandomUtils.java │ │ │ ├── StringPair.java │ │ │ ├── TarUtils.java │ │ │ ├── Utils.java │ │ │ ├── ZipUtils.java │ │ │ ├── cuda │ │ │ ├── CudaLibrary.java │ │ │ ├── CudaUtils.java │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── passthrough │ │ │ ├── PassthroughNDArray.java │ │ │ ├── PassthroughNDManager.java │ │ │ ├── PassthroughTranslator.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ ├── native │ │ └── djl │ │ │ └── utils.h │ ├── resources │ │ ├── META-INF │ │ │ ├── native-image │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── api │ │ │ │ │ ├── native-image.properties │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ └── resource-config.json │ │ │ └── services │ │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ └── ai │ │ │ └── djl │ │ │ └── engine │ │ │ └── api.properties │ └── uml │ │ ├── imperative_model.puml │ │ ├── inference_activity.puml │ │ ├── inference_classes.puml │ │ ├── inference_model_zoo.puml │ │ ├── symbolic_model.puml │ │ ├── training.puml │ │ └── training_activity.puml │ └── test │ ├── java │ └── ai │ │ └── djl │ │ ├── DeviceTest.java │ │ ├── inference │ │ └── streaming │ │ │ ├── ChunkedBytesSupplierTest.java │ │ │ ├── IteratorBytesSupplierTest.java │ │ │ ├── PublisherBytesSupplierTest.java │ │ │ └── package-info.java │ │ ├── metric │ │ ├── MetricsTest.java │ │ └── package-info.java │ │ ├── modality │ │ ├── EnsembleTest.java │ │ ├── InputOutputTest.java │ │ ├── audio │ │ │ ├── AudioFactoryTest.java │ │ │ └── package-info.java │ │ ├── cv │ │ │ ├── output │ │ │ │ ├── CategoryMaskTest.java │ │ │ │ ├── DetectedObjectsTest.java │ │ │ │ ├── RectangleTest.java │ │ │ │ └── package-info.java │ │ │ ├── transform │ │ │ │ ├── PadTest.java │ │ │ │ ├── ResizeShortTest.java │ │ │ │ └── package-info.java │ │ │ └── translator │ │ │ │ ├── BigGANTranslatorFactoryTest.java │ │ │ │ ├── ImageClassificationTranslatorFactoryTest.java │ │ │ │ ├── InstanceSegmentationTranslatorFactoryTest.java │ │ │ │ ├── Sam2InputTest.java │ │ │ │ ├── Sam2TranslatorFactoryTest.java │ │ │ │ ├── SemanticSegmentationTranslatorFactoryTest.java │ │ │ │ ├── SimplePoseTranslatorFactoryTest.java │ │ │ │ ├── SingleShotDetectionTranslatorFactoryTest.java │ │ │ │ ├── StyleTransferTranslatorFactoryTest.java │ │ │ │ ├── YoloTranslatorFactoryTest.java │ │ │ │ ├── YoloV5TranslatorFactoryTest.java │ │ │ │ ├── YoloV8TranslatorFactoryTest.java │ │ │ │ ├── YoloWorldTranslatorTest.java │ │ │ │ └── package-info.java │ │ ├── nlp │ │ │ ├── embedding │ │ │ │ ├── TrainableWordEmbeddingTest.java │ │ │ │ └── package-info.java │ │ │ └── preprocess │ │ │ │ ├── HyphenNormalizerTest.java │ │ │ │ ├── LowerCaseConverterTest.java │ │ │ │ ├── PunctuationSeparatorTest.java │ │ │ │ ├── SimpleTokenizerTest.java │ │ │ │ ├── TextEndpointTokenTest.java │ │ │ │ ├── TextTruncatorTest.java │ │ │ │ ├── UnicodeNormalizerTest.java │ │ │ │ └── package-info.java │ │ └── package-info.java │ │ ├── ndarray │ │ ├── NDListTest.java │ │ ├── NDScopeTest.java │ │ ├── NDSerializerTest.java │ │ ├── NDUtilsTest.java │ │ ├── internal │ │ │ ├── NDFormatTest.java │ │ │ └── package-info.java │ │ ├── package-info.java │ │ └── types │ │ │ ├── DataTypeTest.java │ │ │ ├── ShapeTest.java │ │ │ └── package-info.java │ │ ├── nn │ │ ├── BlockFactoryTest.java │ │ ├── convolutional │ │ │ ├── Conv1dTest.java │ │ │ ├── Conv1dTransposeTest.java │ │ │ ├── Conv2dTest.java │ │ │ ├── Conv2dTransposeTest.java │ │ │ ├── Conv3dTest.java │ │ │ ├── OutputShapeTest.java │ │ │ ├── Range.java │ │ │ ├── ShapeUtils.java │ │ │ ├── TestData.java │ │ │ └── package-info.java │ │ ├── norm │ │ │ ├── GhostBatchNormTest.java │ │ │ └── package-info.java │ │ └── package-info.java │ │ ├── package-info.java │ │ ├── repository │ │ ├── DjlRepositoryTest.java │ │ ├── FuseRepositoryTest.java │ │ ├── JarRepositoryTest.java │ │ ├── TfhubRepositoryTest.java │ │ ├── ZooTest.java │ │ └── package-info.java │ │ ├── training │ │ ├── LearningRateTest.java │ │ ├── package-info.java │ │ └── util │ │ │ ├── MinMaxScalerTest.java │ │ │ └── package-info.java │ │ ├── translate │ │ ├── BatchifierTest.java │ │ ├── DeferredTranslatorFactoryTest.java │ │ ├── NoopServingTranslatorFactoryTest.java │ │ ├── ServingTranslatorFactoryTest.java │ │ ├── ServingTranslatorTest.java │ │ └── package-info.java │ │ └── util │ │ ├── Float16UtilsTest.java │ │ ├── NeuronUtilsTest.java │ │ ├── PairListTest.java │ │ ├── PlatformTest.java │ │ ├── SecurityManagerTest.java │ │ ├── UtilsTest.java │ │ ├── ZipUtilsTest.java │ │ ├── cuda │ │ ├── CudaUtilsTest.java │ │ └── package-info.java │ │ ├── package-info.java │ │ └── passthrough │ │ ├── PassthroughNDManagerTest.java │ │ └── package-info.java │ └── resources │ ├── 0d.npy │ ├── 1d.npy │ ├── 2d.npy │ ├── boolean.npy │ ├── fp16.npy │ ├── fp32.npy │ ├── identity │ └── serving.properties │ ├── int8.npy │ ├── linux-created-offender-root.tar │ ├── linux-created-offender-root.zip │ ├── linux-created-offender-traversal-elements.tar │ ├── linux-created-offender-traversal-elements.zip │ ├── list.npz │ ├── list.safetensors │ ├── offending.tar │ ├── uint8.npy │ ├── windows-created-offender-root.tar │ ├── windows-created-offender-root.zip │ ├── windows-created-offender-traversal-elements.tar │ ├── windows-created-offender-traversal-elements.zip │ └── yolo_world │ ├── identity.pt │ ├── merges.txt │ └── vocab.json ├── apt.txt ├── basicdataset ├── README.md ├── build.gradle.kts ├── gradlew └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── basicdataset │ │ │ ├── BasicDatasets.java │ │ │ ├── cv │ │ │ ├── BananaDetection.java │ │ │ ├── CocoDetection.java │ │ │ ├── CocoMetadata.java │ │ │ ├── CocoUtils.java │ │ │ ├── ImageDataset.java │ │ │ ├── ObjectDetectionDataset.java │ │ │ ├── PikachuDetection.java │ │ │ ├── classification │ │ │ │ ├── AbstractImageFolder.java │ │ │ │ ├── CaptchaDataset.java │ │ │ │ ├── Cifar10.java │ │ │ │ ├── FashionMnist.java │ │ │ │ ├── FruitsFreshAndRotten.java │ │ │ │ ├── ImageClassificationDataset.java │ │ │ │ ├── ImageFolder.java │ │ │ │ ├── ImageNet.java │ │ │ │ ├── Mnist.java │ │ │ │ └── package-info.java │ │ │ └── package-info.java │ │ │ ├── nlp │ │ │ ├── AmazonReview.java │ │ │ ├── CookingStackExchange.java │ │ │ ├── GoEmotions.java │ │ │ ├── PennTreebankText.java │ │ │ ├── StanfordMovieReview.java │ │ │ ├── StanfordQuestionAnsweringDataset.java │ │ │ ├── TatoebaEnglishFrenchDataset.java │ │ │ ├── TextDataset.java │ │ │ ├── UniversalDependenciesEnglishEWT.java │ │ │ ├── WikiText2.java │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── tabular │ │ │ ├── AirfoilRandomAccess.java │ │ │ ├── AmesRandomAccess.java │ │ │ ├── CsvDataset.java │ │ │ ├── DailyDelhiClimate.java │ │ │ ├── ListFeatures.java │ │ │ ├── MapFeatures.java │ │ │ ├── MovieLens100k.java │ │ │ ├── TabularDataset.java │ │ │ ├── TabularResults.java │ │ │ ├── TabularTranslator.java │ │ │ ├── TabularTranslatorFactory.java │ │ │ ├── package-info.java │ │ │ └── utils │ │ │ │ ├── DynamicBuffer.java │ │ │ │ ├── Feature.java │ │ │ │ ├── Featurizer.java │ │ │ │ ├── Featurizers.java │ │ │ │ ├── PreparedFeaturizer.java │ │ │ │ └── package-info.java │ │ │ └── utils │ │ │ ├── FixedBucketSampler.java │ │ │ ├── TextData.java │ │ │ ├── ThrowingFunction.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ └── resources │ │ ├── ai │ │ └── djl │ │ │ └── basicdataset │ │ │ └── tabular │ │ │ └── ames.json │ │ └── imagenet │ │ ├── classes.json │ │ └── extract_imagenet.py │ └── test │ ├── java │ └── ai │ │ └── djl │ │ └── basicdataset │ │ ├── AirfoilRandomAccessTest.java │ │ ├── AmazonReviewsTest.java │ │ ├── AmesRandomAccessTest.java │ │ ├── BananaTest.java │ │ ├── CaptchaTest.java │ │ ├── Cifar10Test.java │ │ ├── CocoTest.java │ │ ├── CookingExchangeTest.java │ │ ├── DailyDelhiClimateTest.java │ │ ├── DatasetUtilsTest.java │ │ ├── FashionMnistTest.java │ │ ├── FixedBucketSamplerTest.java │ │ ├── FruitsFreshAndRottenTest.java │ │ ├── GoEmotionsTest.java │ │ ├── ImageFolderTest.java │ │ ├── ImageNetTest.java │ │ ├── MnistTest.java │ │ ├── MovieLens100kTest.java │ │ ├── PennTreebankTextTest.java │ │ ├── PikachuTest.java │ │ ├── RandomAccessDatasetTest.java │ │ ├── StanfordMovieReviewTest.java │ │ ├── StanfordQuestionAnsweringDatasetTest.java │ │ ├── TatoebaEnglishFrenchDatasetTest.java │ │ ├── TestUtils.java │ │ ├── UniversalDependenciesEnglishEWTTest.java │ │ ├── WikiText2Test.java │ │ ├── package-info.java │ │ └── tabular │ │ ├── TabularTranslatorFactoryTest.java │ │ └── package-info.java │ └── resources │ ├── imagefolder │ ├── cat │ │ └── kitten.jpg │ ├── dog │ │ └── dog_bike_car.jpg │ └── misc │ │ └── pikachu.png │ └── mlrepo │ └── dataset │ ├── cv │ └── ai │ │ └── djl │ │ └── basicdataset │ │ ├── banana │ │ └── metadata.json │ │ ├── captcha │ │ └── metadata.json │ │ ├── cifar10 │ │ └── metadata.json │ │ ├── coco │ │ └── metadata.json │ │ ├── fashmnist │ │ └── metadata.json │ │ ├── mnist │ │ └── metadata.json │ │ ├── pikachu-unittest │ │ └── metadata.json │ │ └── pikachu │ │ └── metadata.json │ ├── nlp │ └── ai │ │ └── djl │ │ └── basicdataset │ │ ├── amazon_reviews │ │ └── metadata.json │ │ ├── cooking_stackexchange │ │ └── metadata.json │ │ ├── goemotions │ │ └── metadata.json │ │ ├── penntreebank-unlabeled-processed │ │ └── metadata.json │ │ ├── stanford-movie-review │ │ └── metadata.json │ │ ├── stanford-question-answer │ │ └── metadata.json │ │ ├── tatoeba-en-fr │ │ └── metadata.json │ │ ├── universal-dependencies │ │ └── universal-dependencies-en-ewt │ │ │ └── metadata.json │ │ └── wikitext-2 │ │ └── metadata.json │ └── tabular │ └── ai │ └── djl │ └── basicdataset │ ├── airfoil │ └── metadata.json │ ├── ames │ └── metadata.json │ ├── daily-delhi-climate │ └── metadata.json │ └── movielens-100k │ └── metadata.json ├── bom ├── README.md ├── build.gradle.kts ├── gradle ├── gradle.properties ├── gradlew └── settings.gradle ├── build.gradle.kts ├── buildSrc ├── build.gradle.kts ├── settings.gradle.kts └── src │ └── main │ └── kotlin │ ├── ai │ └── djl │ │ ├── check.gradle.kts │ │ ├── cppFormatter.gradle.kts │ │ ├── javaBase.gradle.kts │ │ ├── javaFormatter.gradle.kts │ │ ├── javaProject.gradle.kts │ │ ├── publish.gradle.kts │ │ ├── release.gradle.kts │ │ └── stats.gradle.kts │ └── util.kt ├── djl-zero ├── README.md ├── build.gradle.kts ├── gradlew └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── zero │ │ │ ├── Performance.java │ │ │ ├── RequireZoo.java │ │ │ ├── cv │ │ │ ├── ImageClassification.java │ │ │ ├── ObjectDetection.java │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── tabular │ │ │ ├── TabularRegression.java │ │ │ └── package-info.java │ └── javadoc │ │ └── overview.html │ └── test │ └── java │ └── ai │ └── djl │ └── zero │ └── cv │ ├── ImageClassificationTest.java │ └── package-info.java ├── docker ├── README.md ├── spark │ └── Dockerfile ├── tensorrt │ └── Dockerfile └── windows │ └── Dockerfile ├── docs ├── .gitignore ├── README.md ├── blogposts.md ├── create_serving_ready_model.md ├── cv_utils.md ├── d2l.md ├── dataset.md ├── development │ ├── CSVDataset.java │ ├── README.md │ ├── add_dataset_to_djl.md │ ├── add_model_to_model-zoo.md │ ├── cache_management.md │ ├── configure_logging.md │ ├── dependency_management.md │ ├── development_guideline.md │ ├── example_dataset.md │ ├── external_libraries.md │ ├── how_to_use_dataset.md │ ├── img │ │ ├── custom_debug_view.png │ │ ├── ndmanager_structure_for_inference.png │ │ └── ndmanager_structure_for_training.png │ ├── inference_performance_optimization.md │ ├── memory_management.md │ ├── profiler.md │ ├── setup.md │ └── troubleshooting.md ├── engine.md ├── faq.md ├── forums.md ├── get.md ├── how_to_collect_metrics.md ├── hybrid_engine.md ├── interactive_tool.md ├── load_model.md ├── mkdocs.yml ├── model-zoo.md ├── mxnet │ ├── how_to_convert_your_model_to_symbol.md │ └── mxnet_backend_optimizer.md ├── pytorch │ ├── how_to_convert_your_model_to_torchscript.md │ └── pytorch-djl-ndarray-cheatsheet.md ├── quick_start.md ├── roadmap.md ├── telemetry.md └── tensorflow │ └── how_to_import_tensorflow_models_in_DJL.md ├── engines ├── ml │ ├── lightgbm │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── ml │ │ │ │ │ └── lightgbm │ │ │ │ │ ├── LgbmDataset.java │ │ │ │ │ ├── LgbmEngine.java │ │ │ │ │ ├── LgbmEngineProvider.java │ │ │ │ │ ├── LgbmModel.java │ │ │ │ │ ├── LgbmNDArray.java │ │ │ │ │ ├── LgbmNDManager.java │ │ │ │ │ ├── LgbmSymbolBlock.java │ │ │ │ │ ├── jni │ │ │ │ │ ├── JniUtils.java │ │ │ │ │ ├── LibUtils.java │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ └── resources │ │ │ │ └── META-INF │ │ │ │ └── services │ │ │ │ └── ai.djl.engine.EngineProvider │ │ │ └── test │ │ │ └── java │ │ │ └── ai │ │ │ └── djl │ │ │ └── ml │ │ │ └── lightgbm │ │ │ ├── LgbmModelTest.java │ │ │ └── package-info.java │ └── xgboost │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ ├── main │ │ ├── java │ │ │ ├── ai │ │ │ │ └── djl │ │ │ │ │ └── ml │ │ │ │ │ └── xgboost │ │ │ │ │ ├── XgbEngine.java │ │ │ │ │ ├── XgbEngineProvider.java │ │ │ │ │ ├── XgbModel.java │ │ │ │ │ ├── XgbNDArray.java │ │ │ │ │ ├── XgbNDManager.java │ │ │ │ │ ├── XgbSymbolBlock.java │ │ │ │ │ └── package-info.java │ │ │ ├── com │ │ │ │ └── sun │ │ │ │ │ └── jna │ │ │ │ │ ├── PointerProxy.java │ │ │ │ │ └── package-info.java │ │ │ └── ml │ │ │ │ └── dmlc │ │ │ │ └── xgboost4j │ │ │ │ └── java │ │ │ │ ├── JniUtils.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ └── resources │ │ │ └── META-INF │ │ │ └── services │ │ │ └── ai.djl.engine.EngineProvider │ │ └── test │ │ └── java │ │ └── ai │ │ └── djl │ │ └── ml │ │ └── xgboost │ │ ├── XgbModelTest.java │ │ └── package-info.java ├── mxnet │ ├── README.md │ ├── jnarator │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ └── main │ │ │ ├── antlr │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── mxnet │ │ │ │ └── jnarator │ │ │ │ └── parser │ │ │ │ └── C.g4 │ │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── mxnet │ │ │ │ └── jnarator │ │ │ │ ├── AntlrUtils.java │ │ │ │ ├── DataType.java │ │ │ │ ├── FuncInfo.java │ │ │ │ ├── JnaGenerator.java │ │ │ │ ├── JnaParser.java │ │ │ │ ├── Main.java │ │ │ │ ├── Parameter.java │ │ │ │ ├── TypeDefine.java │ │ │ │ └── package-info.java │ │ │ └── resources │ │ │ └── log4j2.xml │ ├── mxnet-engine │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── include │ │ │ │ ├── mxnet │ │ │ │ │ └── c_api.h │ │ │ │ └── nnvm │ │ │ │ │ └── c_api.h │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ ├── engine │ │ │ │ │ ├── CachedOp.java │ │ │ │ │ ├── GradReq.java │ │ │ │ │ ├── MxDataType.java │ │ │ │ │ ├── MxDeviceType.java │ │ │ │ │ ├── MxEngine.java │ │ │ │ │ ├── MxEngineProvider.java │ │ │ │ │ ├── MxGradientCollector.java │ │ │ │ │ ├── MxModel.java │ │ │ │ │ ├── MxNDArray.java │ │ │ │ │ ├── MxNDArrayEx.java │ │ │ │ │ ├── MxNDArrayIndexer.java │ │ │ │ │ ├── MxNDManager.java │ │ │ │ │ ├── MxOpParams.java │ │ │ │ │ ├── MxParameterServer.java │ │ │ │ │ ├── MxSymbolBlock.java │ │ │ │ │ ├── Symbol.java │ │ │ │ │ └── package-info.java │ │ │ │ │ ├── jna │ │ │ │ │ ├── FunctionInfo.java │ │ │ │ │ ├── JnaUtils.java │ │ │ │ │ ├── LibUtils.java │ │ │ │ │ ├── NativeString.java │ │ │ │ │ ├── ObjectPool.java │ │ │ │ │ ├── PointerArray.java │ │ │ │ │ ├── StringArray.java │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ ├── jna │ │ │ │ └── mapping.properties │ │ │ ├── resources │ │ │ │ ├── META-INF │ │ │ │ │ ├── native-image │ │ │ │ │ │ └── ai │ │ │ │ │ │ │ └── djl │ │ │ │ │ │ │ └── mxnet │ │ │ │ │ │ │ └── mxnet-engine │ │ │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ │ │ └── resource-config.json │ │ │ │ │ └── services │ │ │ │ │ │ └── ai.djl.engine.EngineProvider │ │ │ │ └── mxnet-engine.properties │ │ │ └── uml │ │ │ │ ├── engine.puml │ │ │ │ └── mxnet.puml │ │ │ └── test │ │ │ └── java │ │ │ └── ai │ │ │ └── djl │ │ │ └── mxnet │ │ │ └── integration │ │ │ ├── MxBackendOptimizationTest.java │ │ │ ├── MxCoverageTest.java │ │ │ ├── MxGradientCollectorIntegrationTest.java │ │ │ ├── MxParameterStoreTest.java │ │ │ └── package-info.java │ ├── mxnet-model-zoo │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ └── zoo │ │ │ │ │ ├── MxModelZoo.java │ │ │ │ │ ├── MxZooProvider.java │ │ │ │ │ ├── nlp │ │ │ │ │ ├── embedding │ │ │ │ │ │ ├── GloveWordEmbeddingBlockFactory.java │ │ │ │ │ │ ├── GloveWordEmbeddingTranslatorFactory.java │ │ │ │ │ │ ├── package-info.java │ │ │ │ │ │ └── utils │ │ │ │ │ │ │ ├── BuildModelZooWordEmbedding.java │ │ │ │ │ │ │ └── package-info.java │ │ │ │ │ └── qa │ │ │ │ │ │ ├── MxBertQATranslator.java │ │ │ │ │ │ ├── MxBertQATranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ ├── resources │ │ │ │ └── META-INF │ │ │ │ │ ├── native-image │ │ │ │ │ └── ai │ │ │ │ │ │ └── djl │ │ │ │ │ │ └── mxnet │ │ │ │ │ │ └── mxnet-model-zoo │ │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ │ └── resource-config.json │ │ │ │ │ └── services │ │ │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ │ └── scripts │ │ │ │ ├── convertEmbedding.py │ │ │ │ └── exportYolo.py │ │ │ └── test │ │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── mxnet │ │ │ │ └── integration │ │ │ │ ├── MxModelZooCoverageTest.java │ │ │ │ ├── MxSymbolBlockTest.java │ │ │ │ ├── modality │ │ │ │ └── nlp │ │ │ │ │ ├── WordEmbeddingTest.java │ │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ │ └── resources │ │ │ └── mlrepo │ │ │ └── model │ │ │ ├── cv │ │ │ ├── action_recognition │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ └── action_recognition │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ ├── inceptionv3_ucf101-symbol.json │ │ │ │ │ └── vgg16_ucf101-symbol.json │ │ │ │ │ ├── classes.txt │ │ │ │ │ └── metadata.json │ │ │ ├── image_classification │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ ├── alexnet │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── darknet │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── densenet │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── googlenet │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── inceptionv3 │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── mlp │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ └── mlp-symbol.json │ │ │ │ │ ├── metadata.json │ │ │ │ │ └── synset.txt │ │ │ │ │ ├── mobilenet │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── resnest │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── resnet │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ ├── resnet152_v1d-symbol.json │ │ │ │ │ │ ├── resnet18_v1-symbol.json │ │ │ │ │ │ ├── resnet50_v1-symbol.json │ │ │ │ │ │ └── resnet50_v2-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── resnext │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ └── resnext101_64x4d-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── se_resnext │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ ├── se_resnext101_32x4d-symbol.json │ │ │ │ │ │ └── se_resnext101_64x4d-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── senet │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ └── senet_154-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── squeezenet │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ └── squeezenet1.0-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── synset.txt │ │ │ │ │ ├── synset_cifar10.txt │ │ │ │ │ ├── vgg │ │ │ │ │ └── metadata.json │ │ │ │ │ └── xception │ │ │ │ │ └── metadata.json │ │ │ ├── instance_segmentation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ └── mask_rcnn │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ ├── mask_rcnn_resnet101_v1d_coco-symbol.json │ │ │ │ │ └── mask_rcnn_resnet18_v1b_coco-symbol.json │ │ │ │ │ ├── classes.txt │ │ │ │ │ └── metadata.json │ │ │ ├── object_detection │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ ├── classes_coco.txt │ │ │ │ │ ├── classes_voc.txt │ │ │ │ │ ├── ssd │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ │ ├── ssd_512_resnet50_v1-symbol.json │ │ │ │ │ │ └── ssd_512_vgg16_atrous-symbol.json │ │ │ │ │ └── metadata.json │ │ │ │ │ └── yolo │ │ │ │ │ └── metadata.json │ │ │ └── pose_estimation │ │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── mxnet │ │ │ │ └── simple_pose │ │ │ │ ├── 0.0.1 │ │ │ │ ├── simple_pose_resnet101_v1d-symbol.json │ │ │ │ ├── simple_pose_resnet152_v1b-symbol.json │ │ │ │ ├── simple_pose_resnet152_v1d-symbol.json │ │ │ │ ├── simple_pose_resnet18_v1b-symbol.json │ │ │ │ └── simple_pose_resnet50_v1b-symbol.json │ │ │ │ └── metadata.json │ │ │ ├── nlp │ │ │ ├── question_answer │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── mxnet │ │ │ │ │ └── bertqa │ │ │ │ │ ├── 0.0.1 │ │ │ │ │ └── static_bert_qa-symbol.json │ │ │ │ │ └── metadata.json │ │ │ └── word_embedding │ │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── mxnet │ │ │ │ └── glove │ │ │ │ └── metadata.json │ │ │ └── timeseries │ │ │ └── forecasting │ │ │ └── ai │ │ │ └── djl │ │ │ └── mxnet │ │ │ └── deepar │ │ │ └── metadata.json │ └── native │ │ ├── .gitignore │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ └── main │ │ └── resources │ │ └── META-INF │ │ ├── backport.txt │ │ └── native-image │ │ └── ai │ │ └── djl │ │ └── mxnet │ │ └── mxnet-native │ │ └── resource-config.json ├── onnxruntime │ ├── onnxruntime-android │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ └── main │ │ │ └── .empty │ └── onnxruntime-engine │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── onnxruntime │ │ │ │ ├── engine │ │ │ │ ├── OrtEngine.java │ │ │ │ ├── OrtEngineProvider.java │ │ │ │ ├── OrtModel.java │ │ │ │ ├── OrtNDArray.java │ │ │ │ ├── OrtNDManager.java │ │ │ │ ├── OrtSymbolBlock.java │ │ │ │ ├── OrtUtils.java │ │ │ │ └── package-info.java │ │ │ │ └── zoo │ │ │ │ ├── OrtHfModelZoo.java │ │ │ │ ├── OrtHfZooProvider.java │ │ │ │ ├── OrtModelZoo.java │ │ │ │ ├── OrtZooProvider.java │ │ │ │ ├── nlp │ │ │ │ └── textgeneration │ │ │ │ │ ├── OrtGptTranslator.java │ │ │ │ │ ├── OrtGptTranslatorFactory.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ └── tabular │ │ │ │ └── softmax_regression │ │ │ │ ├── IrisClassificationTranslatorFactory.java │ │ │ │ ├── IrisFlower.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ └── resources │ │ │ └── META-INF │ │ │ └── services │ │ │ ├── ai.djl.engine.EngineProvider │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ └── test │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── onnxruntime │ │ │ ├── engine │ │ │ ├── GptTranslatorTest.java │ │ │ ├── OrtTest.java │ │ │ └── package-info.java │ │ │ └── zoo │ │ │ ├── OrtHfModelZooTest.java │ │ │ └── package-info.java │ │ └── resources │ │ └── mlrepo │ │ └── model │ │ ├── cv │ │ ├── image_classification │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── onnxruntime │ │ │ │ └── resnet │ │ │ │ └── metadata.json │ │ ├── instance_segmentation │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── onnxruntime │ │ │ │ ├── yolo11n-seg │ │ │ │ └── metadata.json │ │ │ │ └── yolov8n-seg │ │ │ │ └── metadata.json │ │ ├── object_detection │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── onnxruntime │ │ │ │ ├── yolo11n │ │ │ │ └── metadata.json │ │ │ │ ├── yolo5s │ │ │ │ └── metadata.json │ │ │ │ └── yolov8n │ │ │ │ └── metadata.json │ │ └── pose_estimation │ │ │ └── ai │ │ │ └── djl │ │ │ └── onnxruntime │ │ │ ├── yolo11-pose │ │ │ └── metadata.json │ │ │ └── yolov8n-pose │ │ │ └── metadata.json │ │ └── tabular │ │ └── softmax_regression │ │ └── ai │ │ └── djl │ │ └── onnxruntime │ │ └── iris_flowers │ │ └── metadata.json ├── pytorch │ ├── README.md │ ├── pytorch-engine │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── engine │ │ │ │ │ ├── PtDeviceType.java │ │ │ │ │ ├── PtEngine.java │ │ │ │ │ ├── PtEngineProvider.java │ │ │ │ │ ├── PtGradientCollector.java │ │ │ │ │ ├── PtModel.java │ │ │ │ │ ├── PtNDArray.java │ │ │ │ │ ├── PtNDArrayEx.java │ │ │ │ │ ├── PtNDArrayIndexer.java │ │ │ │ │ ├── PtNDManager.java │ │ │ │ │ ├── PtSymbolBlock.java │ │ │ │ │ └── package-info.java │ │ │ │ │ └── jni │ │ │ │ │ ├── IValue.java │ │ │ │ │ ├── IValueUtils.java │ │ │ │ │ ├── JniUtils.java │ │ │ │ │ ├── LibUtils.java │ │ │ │ │ ├── PyTorchLibrary.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ └── resources │ │ │ │ ├── META-INF │ │ │ │ ├── native-image │ │ │ │ │ └── ai │ │ │ │ │ │ └── djl │ │ │ │ │ │ └── pytorch │ │ │ │ │ │ └── pytorch-engine │ │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ │ └── resource-config.json │ │ │ │ └── services │ │ │ │ │ └── ai.djl.engine.EngineProvider │ │ │ │ └── pytorch-engine.properties │ │ │ └── test │ │ │ └── java │ │ │ └── ai │ │ │ └── djl │ │ │ └── pytorch │ │ │ ├── integration │ │ │ ├── ALibUtilsTest.java │ │ │ ├── IValueTest.java │ │ │ ├── MkldnnTest.java │ │ │ ├── MpsTest.java │ │ │ ├── ProfilerTest.java │ │ │ ├── PtModelTest.java │ │ │ ├── PtNDArrayTest.java │ │ │ ├── TorchScriptTest.java │ │ │ └── package-info.java │ │ │ └── jni │ │ │ ├── IValueUtilsTest.java │ │ │ ├── JniUtilsTest.java │ │ │ └── package-info.java │ ├── pytorch-jni │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ └── main │ │ │ └── resources │ │ │ └── META-INF │ │ │ └── native-image │ │ │ └── ai │ │ │ └── djl │ │ │ └── pytorch │ │ │ └── pytorch-jni │ │ │ └── resource-config.json │ ├── pytorch-model-zoo │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ └── zoo │ │ │ │ │ ├── PtModelZoo.java │ │ │ │ │ ├── PtZooProvider.java │ │ │ │ │ ├── cv │ │ │ │ │ └── objectdetection │ │ │ │ │ │ ├── PtSsdTranslator.java │ │ │ │ │ │ ├── PtSsdTranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ ├── nlp │ │ │ │ │ ├── package-info.java │ │ │ │ │ ├── qa │ │ │ │ │ │ ├── PtBertQATranslator.java │ │ │ │ │ │ ├── PtBertQATranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ ├── sentimentanalysis │ │ │ │ │ │ ├── PtDistilBertTranslator.java │ │ │ │ │ │ ├── PtDistilBertTranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ └── textgeneration │ │ │ │ │ │ ├── PtGptTranslator.java │ │ │ │ │ │ ├── PtGptTranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ └── resources │ │ │ │ └── META-INF │ │ │ │ ├── native-image │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ └── pytorch-model-zoo │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ └── resource-config.json │ │ │ │ └── services │ │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ │ └── test │ │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── pytorch │ │ │ │ └── zoo │ │ │ │ └── nlp │ │ │ │ └── textgeneration │ │ │ │ ├── GptTranslatorTest.java │ │ │ │ └── package-info.java │ │ │ └── resources │ │ │ └── mlrepo │ │ │ └── model │ │ │ ├── cv │ │ │ ├── action_recognition │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ └── Human-Action-Recognition-VIT-Base-patch16-224 │ │ │ │ │ └── metadata.json │ │ │ ├── image-generation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── biggan-deep │ │ │ │ │ └── metadata.json │ │ │ │ │ └── cyclegan │ │ │ │ │ └── metadata.json │ │ │ ├── image_classification │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── resnet │ │ │ │ │ └── metadata.json │ │ │ │ │ └── resnet18_embedding │ │ │ │ │ └── metadata.json │ │ │ ├── instance_segmentation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── yolo11n-seg │ │ │ │ │ └── metadata.json │ │ │ │ │ └── yolov8n-seg │ │ │ │ │ └── metadata.json │ │ │ ├── mask_generation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── sam2-hiera-large-gpu │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-large │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-tiny-gpu │ │ │ │ │ └── metadata.json │ │ │ │ │ └── sam2-hiera-tiny │ │ │ │ │ └── metadata.json │ │ │ ├── object_detection │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── faster_rcnn │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-large-gpu │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-large │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-tiny-gpu │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── sam2-hiera-tiny │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── ssd │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── yolo11n │ │ │ │ │ └── metadata.json │ │ │ │ │ ├── yolov5s │ │ │ │ │ └── metadata.json │ │ │ │ │ └── yolov8n │ │ │ │ │ └── metadata.json │ │ │ ├── pose_estimation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ ├── yolo11n-pose │ │ │ │ │ └── metadata.json │ │ │ │ │ └── yolov8n-pose │ │ │ │ │ └── metadata.json │ │ │ ├── semantic_segmentation │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── pytorch │ │ │ │ │ └── deeplabv3 │ │ │ │ │ └── metadata.json │ │ │ └── zero_shot_object_detection │ │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── pytorch │ │ │ │ └── yolov8s-worldv2 │ │ │ │ └── metadata.json │ │ │ └── nlp │ │ │ ├── question_answer │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── pytorch │ │ │ │ └── bertqa │ │ │ │ └── metadata.json │ │ │ ├── sentiment_analysis │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── pytorch │ │ │ │ └── distilbert │ │ │ │ └── metadata.json │ │ │ └── text_classification │ │ │ └── ai │ │ │ └── djl │ │ │ └── pytorch │ │ │ └── bert │ │ │ └── metadata.json │ └── pytorch-native │ │ ├── .gitignore │ │ ├── CMakeLists.txt │ │ ├── README.md │ │ ├── build.cmd │ │ ├── build.gradle.kts │ │ ├── build.sh │ │ ├── build_android.sh │ │ ├── gradlew │ │ └── src │ │ └── main │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── pytorch │ │ │ └── jni │ │ │ ├── NativeHelper.java │ │ │ └── package-info.java │ │ ├── native │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_inference.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_ivalue.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_nn_functional.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_optim.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_system.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_tensor.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_comparison.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_creation.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_isjm.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_other.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_pointwise.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_random_sampling.cc │ │ ├── ai_djl_pytorch_jni_PyTorchLibrary_torch_reduction.cc │ │ ├── ai_djl_pytorch_jni_cache.cc │ │ ├── ai_djl_pytorch_jni_cache.h │ │ ├── djl_pytorch_jni_exception.h │ │ ├── djl_pytorch_jni_log.cc │ │ ├── djl_pytorch_jni_log.h │ │ └── djl_pytorch_utils.h │ │ ├── patch │ │ ├── 1.11.0 │ │ │ └── Parallel.h │ │ ├── 1.12.1 │ │ │ └── Parallel.h │ │ ├── 1.13.0 │ │ │ └── Parallel.h │ │ ├── 1.13.1 │ │ │ └── Parallel.h │ │ └── cuda.cmake │ │ └── resources │ │ └── META-INF │ │ └── native-image │ │ └── ai │ │ └── djl │ │ └── pytorch │ │ └── pytorch-native │ │ ├── jni-config.json │ │ └── resource-config.json ├── tensorflow │ ├── README.md │ ├── tensorflow-api │ │ ├── .gitignore │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ └── main │ │ │ └── resources │ │ │ └── META-INF │ │ │ └── native-image │ │ │ └── ai │ │ │ └── djl │ │ │ └── tensorflow │ │ │ └── tensorflow-api │ │ │ ├── native-image.properties │ │ │ └── reflect-config.json │ ├── tensorflow-engine │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── tensorflow │ │ │ │ │ └── engine │ │ │ │ │ ├── SavedModelBundle.java │ │ │ │ │ ├── TfDataType.java │ │ │ │ │ ├── TfEngine.java │ │ │ │ │ ├── TfEngineProvider.java │ │ │ │ │ ├── TfModel.java │ │ │ │ │ ├── TfNDArray.java │ │ │ │ │ ├── TfNDArrayEx.java │ │ │ │ │ ├── TfNDArrayIndexer.java │ │ │ │ │ ├── TfNDManager.java │ │ │ │ │ ├── TfOpExecutor.java │ │ │ │ │ ├── TfSymbolBlock.java │ │ │ │ │ ├── javacpp │ │ │ │ │ ├── JavacppUtils.java │ │ │ │ │ ├── LibUtils.java │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ └── resources │ │ │ │ ├── META-INF │ │ │ │ ├── native-image │ │ │ │ │ └── ai │ │ │ │ │ │ └── djl │ │ │ │ │ │ └── tensorflow │ │ │ │ │ │ └── tensorflow-engine │ │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ │ └── resource-config.json │ │ │ │ └── services │ │ │ │ │ └── ai.djl.engine.EngineProvider │ │ │ │ └── tensorflow-engine.properties │ │ │ └── test │ │ │ └── java │ │ │ └── ai │ │ │ └── djl │ │ │ └── tensorflow │ │ │ ├── engine │ │ │ ├── TfNDManagerTest.java │ │ │ └── package-info.java │ │ │ └── integration │ │ │ ├── TfCoverageTest.java │ │ │ └── package-info.java │ ├── tensorflow-model-zoo │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ │ ├── main │ │ │ ├── java │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── tensorflow │ │ │ │ │ └── zoo │ │ │ │ │ ├── TfModelZoo.java │ │ │ │ │ ├── TfZooProvider.java │ │ │ │ │ ├── cv │ │ │ │ │ └── objectdetction │ │ │ │ │ │ ├── TfSsdTranslator.java │ │ │ │ │ │ ├── TfSsdTranslatorFactory.java │ │ │ │ │ │ └── package-info.java │ │ │ │ │ └── package-info.java │ │ │ ├── javadoc │ │ │ │ └── overview.html │ │ │ └── resources │ │ │ │ └── META-INF │ │ │ │ ├── native-image │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── tensorflow │ │ │ │ │ └── tensorflow-model-zoo │ │ │ │ │ ├── reflect-config.json │ │ │ │ │ └── resource-config.json │ │ │ │ └── services │ │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ │ └── test │ │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── tensorflow │ │ │ │ └── integration │ │ │ │ └── modality │ │ │ │ └── cv │ │ │ │ ├── MobileNetTest.java │ │ │ │ ├── ResNetTest.java │ │ │ │ ├── TfSsdTest.java │ │ │ │ └── package-info.java │ │ │ └── resources │ │ │ └── mlrepo │ │ │ └── model │ │ │ └── cv │ │ │ ├── image_classification │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── tensorflow │ │ │ │ ├── mobilenet │ │ │ │ └── metadata.json │ │ │ │ └── resnet │ │ │ │ └── metadata.json │ │ │ └── object_detection │ │ │ └── ai │ │ │ └── djl │ │ │ └── tensorflow │ │ │ └── ssd │ │ │ └── metadata.json │ └── tensorflow-native │ │ ├── README.md │ │ ├── build.gradle.kts │ │ ├── gradlew │ │ └── src │ │ └── main │ │ └── resources │ │ └── META-INF │ │ └── native-image │ │ └── ai │ │ └── djl │ │ └── tensorflow │ │ └── tensorflow-native │ │ └── resource-config.json └── tensorrt │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── build.gradle.kts │ ├── build.sh │ ├── gradlew │ └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── tensorrt │ │ │ ├── engine │ │ │ ├── TrtEngine.java │ │ │ ├── TrtEngineProvider.java │ │ │ ├── TrtModel.java │ │ │ ├── TrtNDArray.java │ │ │ ├── TrtNDManager.java │ │ │ ├── TrtPredictor.java │ │ │ ├── TrtSession.java │ │ │ ├── TrtSymbolBlock.java │ │ │ └── package-info.java │ │ │ └── jni │ │ │ ├── JniUtils.java │ │ │ ├── LibUtils.java │ │ │ ├── TrtLibrary.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ ├── native │ │ ├── ai_djl_tensorrt_jni_TrtLibrary.cc │ │ ├── ai_djl_tensorrt_jni_cache.cc │ │ ├── ai_djl_tensorrt_jni_cache.h │ │ ├── ai_djl_tensorrt_jni_common.h │ │ ├── ai_djl_tensorrt_jni_exception.h │ │ ├── ai_djl_tensorrt_jni_log.cc │ │ ├── ai_djl_tensorrt_jni_log.h │ │ ├── ai_djl_tensorrt_jni_model.cc │ │ └── ai_djl_tensorrt_jni_model.h │ └── resources │ │ ├── META-INF │ │ └── services │ │ │ └── ai.djl.engine.EngineProvider │ │ └── native │ │ └── lib │ │ └── tensorrt.properties │ └── test │ ├── java │ └── ai │ │ └── djl │ │ └── tensorrt │ │ ├── engine │ │ ├── TrtEngineTest.java │ │ ├── TrtNDManagerTest.java │ │ └── package-info.java │ │ └── integration │ │ ├── TrtTest.java │ │ └── package-info.java │ └── resources │ ├── identity.onnx │ ├── identity_70.trt │ └── identity_75.trt ├── examples ├── .gitignore ├── .mvn │ └── wrapper │ │ ├── .gitignore │ │ ├── MavenWrapperDownloader.java │ │ └── maven-wrapper.properties ├── README.md ├── build.gradle.kts ├── docs │ ├── BERT_question_and_answer.md │ ├── action_recognition.md │ ├── biggan.md │ ├── clip_image_text.md │ ├── face_detection.md │ ├── face_recognition.md │ ├── image_classification.md │ ├── img │ │ ├── cat_dog_detected.jpg │ │ ├── detected-dog_bike_car.png │ │ ├── detected-dogs.jpg │ │ ├── detected_instances.png │ │ ├── retinaface_detected.png │ │ └── workFlow.png │ ├── instance_segmentation.md │ ├── mask_detection.md │ ├── neural_machine_translation.md │ ├── object_detection.md │ ├── object_detection_with_tensorflow_saved_model.md │ ├── pose_estimation.md │ ├── segment_anything_2.md │ ├── semantic_segmentation.md │ ├── sentiment_analysis.md │ ├── speech_recognition.md │ ├── stable_diffusion.md │ ├── super_resolution.md │ ├── trace_sam2_img.py │ ├── train_amazon_review_ranking.md │ ├── train_captcha.md │ ├── train_cifar10_resnet.md │ ├── train_mnist_mlp.md │ ├── train_pikachu_ssd.md │ ├── train_transfer_fresh_fruit.md │ └── whisper_speech_text.md ├── gradle ├── gradlew ├── mvnw ├── mvnw.cmd ├── pom.xml └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── examples │ │ │ ├── inference │ │ │ ├── ListModels.java │ │ │ ├── NeuralMachineTranslation.java │ │ │ ├── SpeechRecognition.java │ │ │ ├── clip │ │ │ │ ├── ClipModel.java │ │ │ │ ├── ImageTextComparison.java │ │ │ │ ├── ImageTextTranslator.java │ │ │ │ ├── ImageTranslator.java │ │ │ │ ├── TextTranslator.java │ │ │ │ └── package-info.java │ │ │ ├── cv │ │ │ │ ├── ActionRecognition.java │ │ │ │ ├── BigGAN.java │ │ │ │ ├── ImageClassification.java │ │ │ │ ├── InstanceSegmentation.java │ │ │ │ ├── MaskDetection.java │ │ │ │ ├── ObjectDetection.java │ │ │ │ ├── ObjectDetectionWithTensorflowSavedModel.java │ │ │ │ ├── PoseEstimation.java │ │ │ │ ├── SegmentAnything2.java │ │ │ │ ├── SemanticSegmentation.java │ │ │ │ ├── StyleTransfer.java │ │ │ │ ├── YoloDetection.java │ │ │ │ ├── YoloWorld.java │ │ │ │ ├── ZeroShotImageClassification.java │ │ │ │ ├── ZeroShotObjectDetection.java │ │ │ │ └── package-info.java │ │ │ ├── face │ │ │ │ ├── FaceDetectionTranslator.java │ │ │ │ ├── FeatureComparison.java │ │ │ │ ├── FeatureExtraction.java │ │ │ │ ├── LightFaceDetection.java │ │ │ │ ├── RetinaFaceDetection.java │ │ │ │ └── package-info.java │ │ │ ├── nlp │ │ │ │ ├── BertClassification.java │ │ │ │ ├── BertQaInference.java │ │ │ │ ├── RollingBatch.java │ │ │ │ ├── SentimentAnalysis.java │ │ │ │ ├── SparseRetrieval.java │ │ │ │ ├── TextGeneration.java │ │ │ │ ├── UniversalSentenceEncoder.java │ │ │ │ ├── ZeroShotClassification.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── sr │ │ │ │ ├── SuperResolution.java │ │ │ │ ├── SuperResolutionTranslator.java │ │ │ │ └── package-info.java │ │ │ ├── stablediffusion │ │ │ │ ├── ImageDecoder.java │ │ │ │ ├── ImageEncoder.java │ │ │ │ ├── ImageGeneration.java │ │ │ │ ├── PndmScheduler.java │ │ │ │ ├── StableDiffusionModel.java │ │ │ │ ├── TextEncoder.java │ │ │ │ └── package-info.java │ │ │ ├── timeseries │ │ │ │ ├── AirPassengersDeepAR.java │ │ │ │ ├── M5ForecastingDeepAR.java │ │ │ │ └── package-info.java │ │ │ └── whisper │ │ │ │ ├── SpeechToTextGeneration.java │ │ │ │ ├── WhisperJet.java │ │ │ │ ├── WhisperModel.java │ │ │ │ └── package-info.java │ │ │ └── training │ │ │ ├── TicTacToe.java │ │ │ ├── TrainAirfoilWithTabNet.java │ │ │ ├── TrainBertOnCode.java │ │ │ ├── TrainBertOnGoemotions.java │ │ │ ├── TrainCaptcha.java │ │ │ ├── TrainMnist.java │ │ │ ├── TrainMnistWithLSTM.java │ │ │ ├── TrainPikachu.java │ │ │ ├── TrainPikachuWithYOLOV3.java │ │ │ ├── TrainSentimentAnalysis.java │ │ │ ├── TrainSeq2Seq.java │ │ │ ├── TrainTicTacToe.java │ │ │ ├── TrainTimeSeries.java │ │ │ ├── TrainWithHpo.java │ │ │ ├── TrainWithOptimizers.java │ │ │ ├── package-info.java │ │ │ ├── transferlearning │ │ │ ├── TrainAmazonReviewRanking.java │ │ │ ├── TrainResnetWithCifar10.java │ │ │ ├── TransferFreshFruit.java │ │ │ └── package-info.java │ │ │ └── util │ │ │ ├── Arguments.java │ │ │ ├── BertCodeDataset.java │ │ │ ├── BertGoemotionsDataset.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ ├── python │ │ ├── trace_clip_vit.py │ │ ├── trace_gpt2.py │ │ ├── trace_owlv2.py │ │ └── trace_yolo_worldv2.py │ └── resources │ │ └── log4j2.xml │ └── test │ ├── java │ └── ai │ │ └── djl │ │ ├── examples │ │ ├── inference │ │ │ ├── ListModelsTest.java │ │ │ ├── SpeechRecognitionTest.java │ │ │ ├── clip │ │ │ │ ├── ClipModelTest.java │ │ │ │ └── package-info.java │ │ │ ├── cv │ │ │ │ ├── ActionRecognitionTest.java │ │ │ │ ├── BigGANTest.java │ │ │ │ ├── InstanceSegmentationTest.java │ │ │ │ ├── MaskDetectionTest.java │ │ │ │ ├── ObjectDetectionTest.java │ │ │ │ ├── ObjectDetectionWithTensorflowSavedModelTest.java │ │ │ │ ├── PoseEstimationTest.java │ │ │ │ ├── SegmentAnything2Test.java │ │ │ │ ├── StyleTransferTest.java │ │ │ │ ├── YoloDetectionTest.java │ │ │ │ └── package-info.java │ │ │ ├── face │ │ │ │ ├── FeatureComparisonTest.java │ │ │ │ ├── FeatureExtractionTest.java │ │ │ │ ├── LightFaceDetectionTest.java │ │ │ │ ├── RetinaFaceDetectionTest.java │ │ │ │ └── package-info.java │ │ │ ├── nlp │ │ │ │ ├── BertQaTest.java │ │ │ │ ├── SentimentAnalysisTest.java │ │ │ │ ├── TextGenerationTest.java │ │ │ │ ├── UniversalSentenceEncoderTest.java │ │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ ├── sr │ │ │ │ ├── SuperResolutionTest.java │ │ │ │ └── package-info.java │ │ │ └── timeseries │ │ │ │ ├── TimeSeriesTest.java │ │ │ │ └── package-info.java │ │ └── training │ │ │ ├── TrainAirfoilWithTabNetTest.java │ │ │ ├── TrainAmazonReviewTest.java │ │ │ ├── TrainBertOnGoemotionsTest.java │ │ │ ├── TrainBertTest.java │ │ │ ├── TrainCaptchaTest.java │ │ │ ├── TrainMnistTest.java │ │ │ ├── TrainMnistWithLSTMTest.java │ │ │ ├── TrainPikachuTest.java │ │ │ ├── TrainResNetTest.java │ │ │ ├── TrainSentimentAnalysisTest.java │ │ │ ├── TrainSeq2SeqTest.java │ │ │ ├── TrainTicTacToeTest.java │ │ │ ├── TrainTimeSeriesTest.java │ │ │ ├── TransferFreshFruitTest.java │ │ │ └── package-info.java │ │ └── testing │ │ ├── TestRequirements.java │ │ └── package-info.java │ └── resources │ ├── 0.png │ ├── action_discus_throw.png │ ├── airplane1.png │ ├── dog-cat.jpg │ ├── dog_bike_car.jpg │ ├── fox.png │ ├── kana1.jpg │ ├── kana2.jpg │ ├── kitten.jpg │ ├── largest_selfie.jpg │ ├── mountains.png │ ├── pikachu.jpg │ ├── pose_soccer.png │ ├── segmentation.jpg │ ├── source_wrd2idx.json │ ├── target_idx2wrd.json │ ├── yolov8_synset.txt │ └── yolov8_test.jpg ├── extensions ├── audio │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── audio │ │ │ │ ├── FFmpegAudioFactory.java │ │ │ │ ├── dataset │ │ │ │ ├── AudioData.java │ │ │ │ ├── Librispeech.java │ │ │ │ ├── SpeechRecognitionDataset.java │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ ├── processor │ │ │ │ ├── AudioNormalizer.java │ │ │ │ ├── AudioProcessor.java │ │ │ │ ├── LinearSpecgram.java │ │ │ │ ├── LogMelSpectrogram.java │ │ │ │ ├── PadOrTrim.java │ │ │ │ └── package-info.java │ │ │ │ ├── translator │ │ │ │ ├── WhisperTranslator.java │ │ │ │ ├── WhisperTranslatorFactory.java │ │ │ │ └── package-info.java │ │ │ │ └── util │ │ │ │ ├── AudioUtils.java │ │ │ │ └── package-info.java │ │ └── javadoc │ │ │ └── overview.html │ │ └── test │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── audio │ │ │ ├── FFmpegAudioFactoryTest.java │ │ │ ├── dataset │ │ │ ├── LibrispeechTest.java │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── processor │ │ │ ├── AudioProcessorTest.java │ │ │ └── package-info.java │ │ └── resources │ │ └── mlrepo │ │ └── dataset │ │ └── audio │ │ └── ai │ │ └── djl │ │ └── basicdataset │ │ └── librispeech │ │ └── metadata.json ├── aws-ai │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── aws │ │ │ │ └── s3 │ │ │ │ ├── S3Repository.java │ │ │ │ ├── S3RepositoryFactory.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ └── resources │ │ │ ├── META-INF │ │ │ └── services │ │ │ │ └── ai.djl.repository.RepositoryFactory │ │ │ └── ai │ │ │ └── djl │ │ │ └── aws │ │ │ └── sagemaker │ │ │ ├── assume_role_policy.json │ │ │ └── execution_policy.json │ │ └── test │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── aws │ │ │ └── s3 │ │ │ ├── S3RepositoryFactoryTest.java │ │ │ ├── S3RepositoryTest.java │ │ │ └── package-info.java │ │ └── resources │ │ └── log4j2.xml ├── benchmark │ └── README.md ├── fasttext │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── build.gradle.kts │ ├── build.sh │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── fasttext │ │ │ │ ├── FtAbstractBlock.java │ │ │ │ ├── FtModel.java │ │ │ │ ├── FtTrainingConfig.java │ │ │ │ ├── FtTrainingMode.java │ │ │ │ ├── TrainFastText.java │ │ │ │ ├── jni │ │ │ │ ├── FastTextLibrary.java │ │ │ │ ├── FtWrapper.java │ │ │ │ ├── LibUtils.java │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ └── zoo │ │ │ │ ├── FtModelZoo.java │ │ │ │ ├── FtZooProvider.java │ │ │ │ ├── nlp │ │ │ │ ├── textclassification │ │ │ │ │ ├── FtTextClassification.java │ │ │ │ │ ├── TextClassificationModelLoader.java │ │ │ │ │ └── package-info.java │ │ │ │ └── word_embedding │ │ │ │ │ ├── FtWord2VecWordEmbedding.java │ │ │ │ │ ├── FtWordEmbeddingBlock.java │ │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ ├── native │ │ │ └── ai_djl_fasttext_jni_FastTextLibrary.cc │ │ └── resources │ │ │ ├── META-INF │ │ │ ├── native-image │ │ │ │ └── ai │ │ │ │ │ └── djl │ │ │ │ │ └── fasttext │ │ │ │ │ └── fasttext │ │ │ │ │ ├── jni-config.json │ │ │ │ │ ├── native-image.properties │ │ │ │ │ └── resource-config.json │ │ │ └── services │ │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ │ └── native │ │ │ └── lib │ │ │ └── fasttext.properties │ │ └── test │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── fasttext │ │ │ ├── CookingStackExchangeTest.java │ │ │ └── package-info.java │ │ └── resources │ │ └── mlrepo │ │ └── model │ │ └── nlp │ │ └── text_classification │ │ └── ai │ │ └── djl │ │ └── fasttext │ │ └── cooking_stackexchange │ │ └── metadata.json ├── hadoop │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── hadoop │ │ │ │ └── hdfs │ │ │ │ ├── HdfsRepository.java │ │ │ │ ├── HdfsRepositoryFactory.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ └── resources │ │ │ └── META-INF │ │ │ └── services │ │ │ └── ai.djl.repository.RepositoryFactory │ │ └── test │ │ └── java │ │ └── ai │ │ └── djl │ │ └── hadoop │ │ └── hdfs │ │ ├── HdfsRepositoryFactoryTest.java │ │ ├── HdfsRepositoryTest.java │ │ └── package-info.java ├── opencv │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── opencv │ │ │ │ ├── OpenCVImage.java │ │ │ │ ├── OpenCVImageFactory.java │ │ │ │ └── package-info.java │ │ └── javadoc │ │ │ └── overview.html │ │ └── test │ │ └── java │ │ └── ai │ │ └── djl │ │ └── opencv │ │ ├── OpenCVImageFactoryTest.java │ │ └── package-info.java ├── sentencepiece │ ├── .gitignore │ ├── CMakeLists.txt │ ├── README.md │ ├── build.cmd │ ├── build.gradle.kts │ ├── build.sh │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── sentencepiece │ │ │ │ ├── SpProcessor.java │ │ │ │ ├── SpTextEmbedding.java │ │ │ │ ├── SpTokenizer.java │ │ │ │ ├── SpVocabulary.java │ │ │ │ ├── jni │ │ │ │ ├── LibUtils.java │ │ │ │ ├── SentencePieceLibrary.java │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ ├── javadoc │ │ │ └── overview.html │ │ ├── native │ │ │ └── ai_djl_sentencepiece_jni_SentencePieceLibrary.cc │ │ └── resources │ │ │ ├── META-INF │ │ │ └── native-image │ │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── sentencepiece │ │ │ │ └── sentencepiece │ │ │ │ ├── jni-config.json │ │ │ │ ├── native-image.properties │ │ │ │ └── resource-config.json │ │ │ └── native │ │ │ └── lib │ │ │ └── sentencepiece.properties │ │ └── test │ │ └── java │ │ └── ai │ │ └── djl │ │ └── sentencepiece │ │ ├── SpTextEmbeddingTest.java │ │ ├── SpTokenizerTest.java │ │ ├── SpVocabularyTest.java │ │ └── package-info.java ├── spark │ ├── .gitignore │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ ├── setup │ │ ├── PyPiDescription.rst │ │ ├── __init__.py │ │ ├── djl_spark │ │ │ ├── __init__.py │ │ │ ├── task │ │ │ │ ├── __init__.py │ │ │ │ ├── audio │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── speech_recognizer.py │ │ │ │ │ └── whisper_speech_recognizer.py │ │ │ │ ├── binary │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── binary_predictor.py │ │ │ │ ├── text │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── question_answerer.py │ │ │ │ │ ├── text2text_generator.py │ │ │ │ │ ├── text_classifier.py │ │ │ │ │ ├── text_decoder.py │ │ │ │ │ ├── text_embedder.py │ │ │ │ │ ├── text_encoder.py │ │ │ │ │ ├── text_generator.py │ │ │ │ │ └── text_tokenizer.py │ │ │ │ └── vision │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── image_classifier.py │ │ │ │ │ ├── image_embedder.py │ │ │ │ │ ├── instance_segmenter.py │ │ │ │ │ ├── object_detector.py │ │ │ │ │ └── semantic_segmenter.py │ │ │ └── util │ │ │ │ ├── __init__.py │ │ │ │ ├── dependency_util.py │ │ │ │ └── np_util.py │ │ ├── pyproject.toml │ │ └── setup.py │ └── src │ │ └── main │ │ └── scala │ │ └── ai │ │ └── djl │ │ └── spark │ │ ├── ModelLoader.scala │ │ ├── task │ │ ├── BasePredictor.scala │ │ ├── audio │ │ │ ├── BaseAudioPredictor.scala │ │ │ └── SpeechRecognizer.scala │ │ ├── binary │ │ │ └── BinaryPredictor.scala │ │ ├── text │ │ │ ├── BaseTextPredictor.scala │ │ │ ├── QuestionAnswerer.scala │ │ │ ├── TextClassifier.scala │ │ │ ├── TextDecoder.scala │ │ │ ├── TextEmbedder.scala │ │ │ ├── TextEncoder.scala │ │ │ └── TextTokenizer.scala │ │ └── vision │ │ │ ├── BaseImagePredictor.scala │ │ │ ├── ImageClassifier.scala │ │ │ ├── ImageEmbedder.scala │ │ │ ├── InstanceSegmenter.scala │ │ │ ├── ObjectDetector.scala │ │ │ └── SemanticSegmenter.scala │ │ └── translator │ │ └── binary │ │ ├── NpBinaryTranslator.scala │ │ └── NpBinaryTranslatorFactory.scala ├── tablesaw │ ├── README.md │ ├── build.gradle.kts │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── tablesaw │ │ │ │ ├── TablesawDataset.java │ │ │ │ └── package-info.java │ │ └── javadoc │ │ │ └── overview.html │ │ └── test │ │ └── java │ │ └── ai │ │ └── djl │ │ └── tablesaw │ │ ├── TablesawDatasetTest.java │ │ └── package-info.java ├── timeseries │ ├── README.md │ ├── build.gradle.kts │ ├── docs │ │ └── forecast_with_M5_data.md │ ├── gradlew │ └── src │ │ ├── main │ │ ├── java │ │ │ └── ai │ │ │ │ └── djl │ │ │ │ └── timeseries │ │ │ │ ├── Forecast.java │ │ │ │ ├── SampleForecast.java │ │ │ │ ├── TimeSeriesData.java │ │ │ │ ├── block │ │ │ │ ├── FeatureEmbedder.java │ │ │ │ ├── FeatureEmbedding.java │ │ │ │ ├── MeanScaler.java │ │ │ │ ├── NopScaler.java │ │ │ │ ├── Scaler.java │ │ │ │ └── package-info.java │ │ │ │ ├── dataset │ │ │ │ ├── CsvTimeSeriesDataset.java │ │ │ │ ├── FieldName.java │ │ │ │ ├── M5Forecast.java │ │ │ │ ├── TimeFeaturizer.java │ │ │ │ ├── TimeFeaturizers.java │ │ │ │ ├── TimeSeriesDataset.java │ │ │ │ └── package-info.java │ │ │ │ ├── distribution │ │ │ │ ├── AffineTransformed.java │ │ │ │ ├── Distribution.java │ │ │ │ ├── DistributionLoss.java │ │ │ │ ├── NegativeBinomial.java │ │ │ │ ├── StudentT.java │ │ │ │ ├── output │ │ │ │ │ ├── ArgProj.java │ │ │ │ │ ├── DistributionOutput.java │ │ │ │ │ ├── NegativeBinomialOutput.java │ │ │ │ │ ├── StudentTOutput.java │ │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ │ │ ├── evaluator │ │ │ │ ├── Rmsse.java │ │ │ │ └── package-info.java │ │ │ │ ├── model │ │ │ │ └── deepar │ │ │ │ │ ├── DeepARNetwork.java │ │ │ │ │ ├── DeepARPredictionNetwork.java │ │ │ │ │ ├── DeepARTrainingNetwork.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ ├── timefeature │ │ │ │ ├── Lag.java │ │ │ │ ├── TimeFeature.java │ │ │ │ ├── TimeOffset.java │ │ │ │ └── package-info.java │ │ │ │ ├── transform │ │ │ │ ├── ExpectedNumInstanceSampler.java │ │ │ │ ├── InstanceSampler.java │ │ │ │ ├── PredictionSplitSampler.java │ │ │ │ ├── TimeSeriesTransform.java │ │ │ │ ├── convert │ │ │ │ │ ├── AsArray.java │ │ │ │ │ ├── Convert.java │ │ │ │ │ ├── VstackFeatures.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── feature │ │ │ │ │ ├── AddAgeFeature.java │ │ │ │ │ ├── AddObservedValuesIndicator.java │ │ │ │ │ ├── AddTimeFeature.java │ │ │ │ │ ├── Feature.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── field │ │ │ │ │ ├── Field.java │ │ │ │ │ ├── RemoveFields.java │ │ │ │ │ ├── SelectField.java │ │ │ │ │ ├── SetField.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ └── split │ │ │ │ │ ├── InstanceSplit.java │ │ │ │ │ ├── Split.java │ │ │ │ │ └── package-info.java │ │ │ │ └── translator │ │ │ │ ├── BaseTimeSeriesTranslator.java │ │ │ │ ├── DeepARTranslator.java │ │ │ │ ├── DeepARTranslatorFactory.java │ │ │ │ ├── TransformerTranslator.java │ │ │ │ ├── TransformerTranslatorFactory.java │ │ │ │ └── package-info.java │ │ └── resources │ │ │ └── ai │ │ │ └── djl │ │ │ └── timeseries │ │ │ └── dataset │ │ │ └── m5forecast_parser.json │ │ └── test │ │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── timeseries │ │ │ ├── block │ │ │ ├── BlockTest.java │ │ │ └── package-info.java │ │ │ ├── dataset │ │ │ ├── M5ForecastTest.java │ │ │ └── package-info.java │ │ │ ├── distribution │ │ │ ├── DistributionTest.java │ │ │ └── package-info.java │ │ │ ├── evaluator │ │ │ ├── RmsseTest.java │ │ │ └── package-info.java │ │ │ ├── model │ │ │ ├── DeepARTest.java │ │ │ └── package-info.java │ │ │ ├── timefeature │ │ │ ├── TimeFeatureTest.java │ │ │ └── package-info.java │ │ │ ├── transform │ │ │ ├── TransformTest.java │ │ │ └── package-info.java │ │ │ └── translator │ │ │ ├── DeepARTranslatorTest.java │ │ │ ├── TransformerTranslatorTest.java │ │ │ └── package-info.java │ │ └── resources │ │ └── mlrepo │ │ └── dataset │ │ └── tabular │ │ └── ai │ │ └── djl │ │ └── basicdataset │ │ └── m5forecast-unittest │ │ └── metadata.json └── tokenizers │ ├── .gitignore │ ├── README.md │ ├── build.cmd │ ├── build.gradle.kts │ ├── build.sh │ ├── build_android.sh │ ├── gradlew │ ├── rust │ ├── .gitignore │ ├── Cargo.lock │ ├── Cargo.toml │ └── src │ │ ├── build.rs │ │ ├── compute_cap.rs │ │ ├── layers │ │ ├── cublaslt.rs │ │ ├── layer_norm.rs │ │ ├── linear.rs │ │ ├── mod.rs │ │ └── rms_norm.rs │ │ ├── lib.rs │ │ ├── models │ │ ├── bert.rs │ │ ├── camembert.rs │ │ ├── distilbert.rs │ │ ├── gemma2.rs │ │ ├── gte.rs │ │ ├── mistral.rs │ │ ├── mod.rs │ │ ├── qwen2.rs │ │ ├── roberta.rs │ │ └── xlm_roberta.rs │ │ ├── ndarray │ │ ├── binary.rs │ │ ├── cmp.rs │ │ ├── creation.rs │ │ ├── mod.rs │ │ ├── nn.rs │ │ ├── other.rs │ │ ├── reduce.rs │ │ └── unary.rs │ │ └── utils.rs │ └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ ├── engine │ │ │ └── rust │ │ │ │ ├── RsEngine.java │ │ │ │ ├── RsEngineProvider.java │ │ │ │ ├── RsModel.java │ │ │ │ ├── RsNDArray.java │ │ │ │ ├── RsNDArrayEx.java │ │ │ │ ├── RsNDArrayIndexer.java │ │ │ │ ├── RsNDManager.java │ │ │ │ ├── RsSymbolBlock.java │ │ │ │ ├── RustLibrary.java │ │ │ │ ├── package-info.java │ │ │ │ └── zoo │ │ │ │ ├── RsModelZoo.java │ │ │ │ ├── RsZooProvider.java │ │ │ │ └── package-info.java │ │ │ └── huggingface │ │ │ ├── tokenizers │ │ │ ├── Encoding.java │ │ │ ├── HuggingFaceTokenizer.java │ │ │ ├── jni │ │ │ │ ├── CharSpan.java │ │ │ │ ├── LibUtils.java │ │ │ │ ├── TokenizersLibrary.java │ │ │ │ └── package-info.java │ │ │ └── package-info.java │ │ │ ├── translator │ │ │ ├── CrossEncoderTranslator.java │ │ │ ├── FillMaskTranslator.java │ │ │ ├── FillMaskTranslatorFactory.java │ │ │ ├── PretrainedConfig.java │ │ │ ├── QuestionAnsweringTranslator.java │ │ │ ├── QuestionAnsweringTranslatorFactory.java │ │ │ ├── SparseRetrievalTranslator.java │ │ │ ├── TextClassificationTranslator.java │ │ │ ├── TextClassificationTranslatorFactory.java │ │ │ ├── TextEmbeddingTranslator.java │ │ │ ├── TextEmbeddingTranslatorFactory.java │ │ │ ├── TokenClassificationTranslator.java │ │ │ ├── TokenClassificationTranslatorFactory.java │ │ │ ├── ZeroShotClassificationTranslator.java │ │ │ ├── ZeroShotClassificationTranslatorFactory.java │ │ │ ├── ZeroShotImageClassificationTranslator.java │ │ │ ├── ZeroShotImageClassificationTranslatorFactory.java │ │ │ ├── ZeroShotObjectDetectionTranslator.java │ │ │ ├── ZeroShotObjectDetectionTranslatorFactory.java │ │ │ └── package-info.java │ │ │ └── zoo │ │ │ ├── HfModelZoo.java │ │ │ ├── HfZooProvider.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ ├── python │ │ ├── .gitignore │ │ ├── README.md │ │ ├── djl_converter │ │ │ ├── __init__.py │ │ │ ├── arg_parser.py │ │ │ ├── fill_mask_converter.py │ │ │ ├── huggingface_converter.py │ │ │ ├── huggingface_models.py │ │ │ ├── metadata.py │ │ │ ├── model_converter.py │ │ │ ├── model_zoo_importer.py │ │ │ ├── question_answering_converter.py │ │ │ ├── safetensors_convert.py │ │ │ ├── sentence_similarity_converter.py │ │ │ ├── shasum.py │ │ │ ├── text_classification_converter.py │ │ │ ├── token_classification_converter.py │ │ │ ├── zero_shot_classification_converter.py │ │ │ └── zip_utils.py │ │ ├── pyproject.toml │ │ ├── requirements.txt │ │ └── setup.py │ └── resources │ │ ├── META-INF │ │ └── services │ │ │ ├── ai.djl.engine.EngineProvider │ │ │ └── ai.djl.repository.zoo.ZooProvider │ │ └── native │ │ └── lib │ │ └── tokenizers.properties │ └── test │ ├── java │ └── ai │ │ └── djl │ │ ├── engine │ │ └── rust │ │ │ ├── NDArrayTests.java │ │ │ ├── RsEngineTest.java │ │ │ ├── package-info.java │ │ │ └── zoo │ │ │ ├── RsModelZooTest.java │ │ │ └── package-info.java │ │ └── huggingface │ │ ├── tokenizers │ │ ├── BpeTokenizerBuilderTest.java │ │ ├── CrossEncoderTranslatorTest.java │ │ ├── FillMaskTranslatorTest.java │ │ ├── HuggingFaceTokenizerTest.java │ │ ├── QuestionAnsweringTranslatorTest.java │ │ ├── SparseRetrievalTranslatorTest.java │ │ ├── TextClassificationTranslatorTest.java │ │ ├── TextEmbeddingTranslatorTest.java │ │ ├── TokenClassificationTranslatorTest.java │ │ ├── ZeroShotClassificationTranslatorTest.java │ │ ├── ZeroShotImageClassificationTranslatorTest.java │ │ ├── ZeroShotObjectDetectionTranslatorTest.java │ │ └── package-info.java │ │ └── zoo │ │ ├── ModelZooTest.java │ │ └── package-info.java │ └── resources │ └── fake-tokenizer-with-padding │ └── tokenizer.json ├── gradle.properties ├── gradle ├── libs.versions.toml └── wrapper │ ├── .gitignore │ ├── GradleWrapperDownloader.java │ └── gradle-wrapper.properties ├── gradlew ├── gradlew.bat ├── index.html ├── index1.0.html ├── integration ├── README.md ├── build.gradle.kts ├── gradle ├── gradlew └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── integration │ │ │ ├── IntegrationTest.java │ │ │ ├── package-info.java │ │ │ ├── tests │ │ │ ├── inference │ │ │ │ ├── PredictorTest.java │ │ │ │ ├── StreamingTest.java │ │ │ │ └── package-info.java │ │ │ ├── modality │ │ │ │ ├── cv │ │ │ │ │ ├── BufferedImageFactoryTest.java │ │ │ │ │ ├── MultiBoxDetectionTest.java │ │ │ │ │ ├── MultiBoxPriorTest.java │ │ │ │ │ ├── MultiBoxTargetTest.java │ │ │ │ │ ├── NDImageUtilsTest.java │ │ │ │ │ ├── SingleShotDetectionTest.java │ │ │ │ │ └── package-info.java │ │ │ │ └── nlp │ │ │ │ │ ├── SimpleTextEncoderTest.java │ │ │ │ │ └── package-info.java │ │ │ ├── model_zoo │ │ │ │ ├── CustomTranslatorTest.java │ │ │ │ ├── ImperativeModelZooTest.java │ │ │ │ ├── classification │ │ │ │ │ ├── AlexNetTest.java │ │ │ │ │ ├── GoogLeNetTest.java │ │ │ │ │ ├── LeNetTest.java │ │ │ │ │ ├── MobileNetV1Test.java │ │ │ │ │ ├── MobileNetV2Test.java │ │ │ │ │ ├── NiNTest.java │ │ │ │ │ ├── ResnetTest.java │ │ │ │ │ ├── SqueezenetTest.java │ │ │ │ │ ├── VGGTest.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── object_detection │ │ │ │ │ ├── SingleShotDetectionTest.java │ │ │ │ │ ├── YOLOv3Test.java │ │ │ │ │ └── package-info.java │ │ │ │ ├── package-info.java │ │ │ │ └── tabular │ │ │ │ │ ├── TabNetTest.java │ │ │ │ │ └── package-info.java │ │ │ ├── ndarray │ │ │ │ ├── NDArrayAttachmentTest.java │ │ │ │ ├── NDArrayCreationOpTest.java │ │ │ │ ├── NDArrayElementArithmeticOpTest.java │ │ │ │ ├── NDArrayElementComparisonOpTest.java │ │ │ │ ├── NDArrayLogicalOpTest.java │ │ │ │ ├── NDArrayNumericOpTest.java │ │ │ │ ├── NDArrayOtherOpTest.java │ │ │ │ ├── NDArrayReductionOpTest.java │ │ │ │ ├── NDArrayShapesManipulationOpTest.java │ │ │ │ ├── NDIndexTest.java │ │ │ │ ├── NDListTest.java │ │ │ │ └── package-info.java │ │ │ ├── nn │ │ │ │ ├── BlockCoreTest.java │ │ │ │ ├── BlockFactoryTest.java │ │ │ │ ├── PoolingOperationsTest.java │ │ │ │ ├── ScaledDotProductAttentionBlockTest.java │ │ │ │ └── package-info.java │ │ │ ├── repository │ │ │ │ ├── RepositoryTest.java │ │ │ │ └── package-info.java │ │ │ ├── training │ │ │ │ ├── ActivationTest.java │ │ │ │ ├── BlocksTest.java │ │ │ │ ├── DatasetTest.java │ │ │ │ ├── EvaluateDatasetTest.java │ │ │ │ ├── EvaluatorTest.java │ │ │ │ ├── GradientCollectorIntegrationTest.java │ │ │ │ ├── LossTest.java │ │ │ │ ├── ModelTest.java │ │ │ │ ├── OptimizerTest.java │ │ │ │ ├── WeightDecayTest.java │ │ │ │ ├── listener │ │ │ │ │ ├── EarlyStoppingListenerTest.java │ │ │ │ │ └── package-info.java │ │ │ │ └── package-info.java │ │ │ └── translate │ │ │ │ ├── PaddingStackBatchifierTest.java │ │ │ │ ├── SimplePaddingStackBatchifierTest.java │ │ │ │ ├── StackBatchifierTest.java │ │ │ │ └── package-info.java │ │ │ └── util │ │ │ ├── Arguments.java │ │ │ ├── DebugEnvironment.java │ │ │ ├── TestUtils.java │ │ │ └── package-info.java │ └── resources │ │ └── log4j2.xml │ └── test │ ├── java │ └── ai │ │ └── djl │ │ ├── CoverageTest.java │ │ ├── integration │ │ ├── IntegrationTests.java │ │ └── package-info.java │ │ └── package-info.java │ └── translator │ └── MyTranslator.java ├── jacoco ├── build.gradle.kts └── gradlew ├── jupyter └── README.md ├── leaders.md ├── model-zoo ├── README.md ├── build.gradle.kts ├── gradlew └── src │ ├── main │ ├── java │ │ └── ai │ │ │ └── djl │ │ │ └── basicmodelzoo │ │ │ ├── BasicModelZoo.java │ │ │ ├── BasicZooProvider.java │ │ │ ├── basic │ │ │ ├── Mlp.java │ │ │ ├── MlpBlockFactory.java │ │ │ └── package-info.java │ │ │ ├── cv │ │ │ ├── classification │ │ │ │ ├── AlexNet.java │ │ │ │ ├── GoogLeNet.java │ │ │ │ ├── LeNet.java │ │ │ │ ├── MobileNetV1.java │ │ │ │ ├── MobileNetV2.java │ │ │ │ ├── NiN.java │ │ │ │ ├── ResNetV1.java │ │ │ │ ├── ResnetBlockFactory.java │ │ │ │ ├── SqueezeNet.java │ │ │ │ ├── VGG.java │ │ │ │ └── package-info.java │ │ │ ├── object_detection │ │ │ │ ├── ssd │ │ │ │ │ ├── SingleShotDetection.java │ │ │ │ │ ├── SsdBlockFactory.java │ │ │ │ │ └── package-info.java │ │ │ │ └── yolo │ │ │ │ │ ├── YOLOV3.java │ │ │ │ │ └── package-info.java │ │ │ └── package-info.java │ │ │ ├── nlp │ │ │ ├── SimpleTextDecoder.java │ │ │ ├── SimpleTextEncoder.java │ │ │ └── package-info.java │ │ │ ├── package-info.java │ │ │ └── tabular │ │ │ ├── TabNet.java │ │ │ └── package-info.java │ ├── javadoc │ │ └── overview.html │ └── resources │ │ └── META-INF │ │ ├── native-image │ │ └── ai │ │ │ └── djl │ │ │ └── basicmodelzoo │ │ │ ├── reflect-config.json │ │ │ └── resource-config.json │ │ └── services │ │ └── ai.djl.repository.zoo.ZooProvider │ └── test │ └── resources │ └── mlrepo │ └── model │ └── cv │ ├── image_classification │ └── ai │ │ └── djl │ │ └── zoo │ │ ├── mlp │ │ └── metadata.json │ │ ├── resnet │ │ └── metadata.json │ │ ├── synset_cifar10.txt │ │ ├── synset_imagenet.txt │ │ └── synset_mnist.txt │ └── object_detection │ └── ai │ └── djl │ └── zoo │ ├── ssd │ └── metadata.json │ └── synset_pikachu.txt ├── postBuild ├── serving └── README.md ├── settings.gradle.kts ├── testing ├── README.md ├── build.gradle.kts ├── gradle ├── gradlew └── src │ └── main │ └── java │ └── ai │ └── djl │ └── testing │ ├── Assertions.java │ ├── CoverageUtils.java │ ├── TestRequirements.java │ └── package-info.java ├── tools ├── conf │ ├── checkstyle.xml │ ├── findbugs-exclude.xml │ ├── licenseHeader.java │ ├── pmd.xml │ └── suppressions.xml ├── gradle │ └── android-formatter.gradle └── scripts │ ├── add_online_runner.py │ ├── broken_link_checker.sh │ └── build-website.sh └── website ├── blog.html ├── css ├── style.css └── style1.0.css ├── img ├── DJL.svg ├── background2.png ├── bg │ ├── img-bg-customer.jpg │ ├── img-bg-feature.jpeg │ ├── img-bg-sky.jpg │ ├── img-bg-twitter.jpg │ ├── img-bg5.jpg │ └── img-customer-bg.jpg ├── community_leader │ ├── anthony.jpeg │ ├── aziz.jpg │ ├── christoph.png │ ├── erik.png │ ├── frank.png │ ├── fyz.jpg │ ├── jake.jpg │ ├── keerthan.jpg │ ├── kimi.png │ ├── lai.jpg │ ├── lu.png │ ├── marat.png │ ├── marcos.png │ ├── qing.jpeg │ ├── stanislav.png │ ├── wei.jpg │ └── zach.jpeg ├── deepjavalibrary.png ├── djl-large.png ├── djl-middle.png ├── djl.png ├── favicon.png └── stars-in-the-night-sky.jpg ├── javadoc.html └── js ├── blockrunner.js ├── doodle_draw.js ├── index.js ├── init.js ├── script.js └── terminal.js /.devcontainer/library-scripts/README.md: -------------------------------------------------------------------------------- 1 | # Warning: Folder contents may be replaced 2 | 3 | The contents of this folder will be automatically replaced with a file of the same name in the [vscode-dev-containers](https://github.com/microsoft/vscode-dev-containers) repository's [script-library folder](https://github.com/microsoft/vscode-dev-containers/tree/master/script-library) whenever the repository is packaged. 4 | 5 | To retain your edits, move the file to a different location. You may also delete the files if they are not needed. -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- 1 | # Global Owners 2 | * @zachgk @frankfliu @DeepJavaLibrary/djl-admin -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- 1 | --- 2 | name: Enhancement 3 | about: Suggest an enhancement, feature request, or idea for this project 4 | title: '' 5 | labels: 'enhancement' 6 | assignees: '' 7 | 8 | --- 9 | 10 | ## Description 11 | 12 | (A clear and concise description of what the feature is.) 13 | 14 | Will this change the current api? How? 15 | 16 | Who will benefit from this enhancement? 17 | 18 | ## References 19 | 20 | - list reference and related literature 21 | - list known implementations 22 | -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- 1 | ## Description ## 2 | 3 | Brief description of what this PR is about 4 | 5 | - If this change is a backward incompatible change, why must this change be made? 6 | - Interesting edge cases to note here 7 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | .gradle 2 | .DS_Store 3 | .idea 4 | .clang 5 | *.iml 6 | .ipynb_checkpoints 7 | build 8 | libs 9 | *.gz 10 | *.params 11 | *.zip 12 | *.jar 13 | *.so 14 | *.dylib 15 | *.dll 16 | *.class 17 | *.log 18 | *.onnx 19 | *.pt 20 | *.safetensors 21 | *.pb 22 | *.trt 23 | *.uff 24 | *.gguf 25 | *.xgb 26 | *.bin 27 | 28 | # Eclipse 29 | .settings 30 | .project 31 | .classpath 32 | bin/ 33 | 34 | #node 35 | node_modules/ 36 | **/static/dist/ 37 | 38 | #vscode 39 | .vscode 40 | -------------------------------------------------------------------------------- /CNAME: -------------------------------------------------------------------------------- 1 | djl.ai -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- 1 | # Code of Conduct 2 | 3 | This project has adopted the [Amazon Open Source Code of Conduct](https://aws.github.io/code-of-conduct). 4 | For more information see the [Code of Conduct FAQ](https://aws.github.io/code-of-conduct-faq) or contact 5 | opensource-codeofconduct@amazon.com with any additional questions or comments. 6 | -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /android/.gitignore: -------------------------------------------------------------------------------- 1 | local.properties 2 | captures 3 | .externalNativeBuild 4 | .cxx 5 | -------------------------------------------------------------------------------- /android/core/consumer-rules.pro: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/android/core/consumer-rules.pro -------------------------------------------------------------------------------- /android/core/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /android/core/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /android/gradle: -------------------------------------------------------------------------------- 1 | ../gradle -------------------------------------------------------------------------------- /android/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /android/gradlew.bat: -------------------------------------------------------------------------------- 1 | call ..\gradlew.bat -------------------------------------------------------------------------------- /android/pytorch-native/consumer-rules.pro: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/android/pytorch-native/consumer-rules.pro -------------------------------------------------------------------------------- /android/pytorch-native/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /android/pytorch-native/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /android/settings.gradle: -------------------------------------------------------------------------------- 1 | rootProject.name='android' 2 | include ':core' 3 | include ':pytorch-native' 4 | include ':tokenizer-native' 5 | -------------------------------------------------------------------------------- /android/tokenizer-native/README.md: -------------------------------------------------------------------------------- 1 | # Tokenizer 2 | 3 | This folder contains build of tokenizer-native library for Android. 4 | -------------------------------------------------------------------------------- /android/tokenizer-native/consumer-rules.pro: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/android/tokenizer-native/consumer-rules.pro -------------------------------------------------------------------------------- /android/tokenizer-native/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /android/tokenizer-native/src/main/AndroidManifest.xml: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /api/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/audio/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for audio processing. */ 15 | package ai.djl.modality.audio; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/audio/translator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for speech recognition processing. */ 15 | package ai.djl.modality.audio.translator; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/cv/output/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains output types used in various computer vision applications. */ 15 | package ai.djl.modality.cv.output; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/cv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for computer vision tasks and image processing. */ 15 | package ai.djl.modality.cv; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/cv/translator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains translators used for computer vision applications. */ 15 | package ai.djl.modality.cv.translator; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/cv/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for image manipulation. */ 15 | package ai.djl.modality.cv.util; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/bert/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains classes that deal with BERT for natural language pre-processing tasks. */ 14 | package ai.djl.modality.nlp.bert; 15 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/generate/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for image manipulation. */ 15 | package ai.djl.modality.nlp.generate; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains utility classes for natural language processing tasks. */ 14 | package ai.djl.modality.nlp; 15 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/preprocess/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains utility classes for natural language pre-processing tasks. */ 14 | package ai.djl.modality.nlp.preprocess; 15 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/qa/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains utility classes for question and answer processing. */ 14 | package ai.djl.modality.nlp.qa; 15 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/nlp/translator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for each of the predefined translator. */ 15 | package ai.djl.modality.nlp.translator; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for each of the predefined modalities. */ 15 | package ai.djl.modality; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/modality/rl/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utility classes for reinforcement learning. */ 15 | package ai.djl.modality.rl; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/ndarray/internal/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains internal helpers for {@link ai.djl.ndarray.NDArray}. */ 15 | package ai.djl.ndarray.internal; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/ndarray/types/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes that define n-dimensional array data types. */ 15 | package ai.djl.ndarray.types; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/nn/core/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes that define simple neural network operations. */ 15 | package ai.djl.nn.core; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/nn/norm/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes that define normalizing neural network operations. */ 15 | package ai.djl.nn.norm; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/nn/transformer/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains blocks for transformer models. */ 15 | package ai.djl.nn.transformer; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/training/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities to use during training. */ 15 | package ai.djl.training.util; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/util/cuda/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities to access CUDA driver. */ 15 | package ai.djl.util.cuda; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities used throughout the API. */ 15 | package ai.djl.util; 16 | -------------------------------------------------------------------------------- /api/src/main/java/ai/djl/util/passthrough/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains passthrough DJL classes for use in extensions and hybrid engines. */ 15 | package ai.djl.util.passthrough; 16 | -------------------------------------------------------------------------------- /api/src/main/resources/META-INF/native-image/ai/djl/api/native-image.properties: -------------------------------------------------------------------------------- 1 | Args = -H:ReflectionConfigurationResources=${.}/reflect-config.json \ 2 | -H:ResourceConfigurationResources=${.}/resource-config.json \ 3 | --initialize-at-run-time=ai.djl.engine.Engine,ai.djl.repository.RepositoryFactoryImpl 4 | -------------------------------------------------------------------------------- /api/src/main/resources/META-INF/native-image/ai/djl/api/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.repository.zoo.ZooProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /api/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.repository.zoo.DefaultZooProvider 2 | -------------------------------------------------------------------------------- /api/src/main/resources/ai/djl/engine/api.properties: -------------------------------------------------------------------------------- 1 | djl_version=${version} -------------------------------------------------------------------------------- /api/src/main/uml/imperative_model.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> [An Engine is associated with the Model] "Create Model" 3 | --> "Build neural network" 4 | --> [model.setBlock()] "Set neural network" 5 | --> [model.load() on specified device)] "Load pre-trained parameters" 6 | --> [model.newPredictor] "Create predictor" 7 | --> [predictor.predict()] "Inference" 8 | --> [model.close()] (*) 9 | @enduml 10 | -------------------------------------------------------------------------------- /api/src/main/uml/inference_activity.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> "Load your pre-trained Model" 3 | --> "Create Translator(pre-process/post-process)" 4 | --> "Create Predictor with Translator" 5 | --> "Inference: predictor.predict(input)" 6 | (*) --> "Prepare input" 7 | "Prepare input" --> "Inference: predictor.predict(input)" 8 | --> "Get prediction result" 9 | --> (*) 10 | @enduml 11 | -------------------------------------------------------------------------------- /api/src/main/uml/inference_model_zoo.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> "Load Model form model zoo" 3 | --> "Create Predictor" 4 | --> "Inference: predictor.predict(input)" 5 | (*) --> "Prepare input" 6 | "Prepare input" --> "Inference: predictor.predict(input)" 7 | --> "Get prediction result" 8 | --> (*) 9 | @enduml 10 | -------------------------------------------------------------------------------- /api/src/main/uml/symbolic_model.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> [An Engine is associated with the Model] "Create Model" 3 | --> [model.load() on specified device)] "Load neural network and pre-trained parameters" 4 | --> [model.newPredictor] "Create predictor" 5 | --> [predictor.predict()] "Inference" 6 | --> [model.close()] (*) 7 | @enduml 8 | -------------------------------------------------------------------------------- /api/src/main/uml/training.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> [An Engine is associated with the Model] "Create Model" 3 | --> "Build neural network" 4 | --> [model.setBlock()] "Set neural network" 5 | --> [model.newTrainer() on specified device(s)] "Create trainer" 6 | --> [trainer.fit()] "Train the model" 7 | --> [model.save()] "Save the model" 8 | --> [model.close()] (*) 9 | @enduml 10 | -------------------------------------------------------------------------------- /api/src/main/uml/training_activity.puml: -------------------------------------------------------------------------------- 1 | @startuml 2 | (*) --> "Build neural network" 3 | --> "Create Model with neural network" 4 | --> "Determined training configuration" 5 | --> "Create Trainer with TrainingConfig" 6 | --> "Train model: trainer.fit(dataset)" 7 | (*) --> "Prepare training dataset" 8 | "Prepare training dataset" --> "Train model: trainer.fit(dataset)" 9 | --> "Save Model" 10 | --> (*) 11 | @enduml 12 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/inference/streaming/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.inference.streaming}. */ 14 | package ai.djl.inference.streaming; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/metric/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.metric}. */ 15 | package ai.djl.metric; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/audio/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the audio api. */ 15 | package ai.djl.modality.audio; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/cv/output/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.modality.cv.output}. */ 14 | package ai.djl.modality.cv.output; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/cv/transform/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2025 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.modality.cv.transform}. */ 14 | package ai.djl.modality.cv.transform; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/cv/translator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.modality.cv.translator}. */ 14 | package ai.djl.modality.cv.translator; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/nlp/embedding/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains test for NLP embeddings. */ 14 | package ai.djl.modality.nlp.embedding; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/nlp/preprocess/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains test for NLP preprocessors. */ 14 | package ai.djl.modality.nlp.preprocess; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/modality/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.modality}. */ 14 | package ai.djl.modality; 15 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/ndarray/internal/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.ndarray.internal}. */ 15 | package ai.djl.ndarray.internal; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/ndarray/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.ndarray}. */ 15 | package ai.djl.ndarray; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/ndarray/types/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.ndarray.types}. */ 15 | package ai.djl.ndarray.types; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/nn/convolutional/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.nn.convolutional}. */ 15 | package ai.djl.nn.convolutional; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/nn/norm/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.nn.norm}. */ 15 | package ai.djl.nn.norm; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/nn/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.nn}. */ 15 | package ai.djl.nn; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the api. */ 15 | package ai.djl; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/repository/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.repository}. */ 15 | package ai.djl.repository; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/training/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.training}. */ 15 | package ai.djl.training; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/training/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.training.util}. */ 15 | package ai.djl.training.util; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/translate/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.translate}. */ 15 | package ai.djl.translate; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/util/cuda/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.util.cuda}. */ 15 | package ai.djl.util.cuda; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.util}. */ 15 | package ai.djl.util; 16 | -------------------------------------------------------------------------------- /api/src/test/java/ai/djl/util/passthrough/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.util.passthrough}. */ 15 | package ai.djl.util.passthrough; 16 | -------------------------------------------------------------------------------- /api/src/test/resources/0d.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/0d.npy -------------------------------------------------------------------------------- /api/src/test/resources/1d.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/1d.npy -------------------------------------------------------------------------------- /api/src/test/resources/2d.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/2d.npy -------------------------------------------------------------------------------- /api/src/test/resources/boolean.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/boolean.npy -------------------------------------------------------------------------------- /api/src/test/resources/fp16.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/fp16.npy -------------------------------------------------------------------------------- /api/src/test/resources/fp32.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/fp32.npy -------------------------------------------------------------------------------- /api/src/test/resources/identity/serving.properties: -------------------------------------------------------------------------------- 1 | blockFactory=ai.djl.nn.IdentityBlockFactory 2 | option.hasParameter=false 3 | translatorFactory=ai.djl.modality.cv.translator.ImageClassificationTranslatorFactory 4 | -------------------------------------------------------------------------------- /api/src/test/resources/int8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/int8.npy -------------------------------------------------------------------------------- /api/src/test/resources/linux-created-offender-root.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/linux-created-offender-root.zip -------------------------------------------------------------------------------- /api/src/test/resources/linux-created-offender-traversal-elements.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/linux-created-offender-traversal-elements.zip -------------------------------------------------------------------------------- /api/src/test/resources/list.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/list.npz -------------------------------------------------------------------------------- /api/src/test/resources/list.safetensors: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/list.safetensors -------------------------------------------------------------------------------- /api/src/test/resources/offending.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/offending.tar -------------------------------------------------------------------------------- /api/src/test/resources/uint8.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/uint8.npy -------------------------------------------------------------------------------- /api/src/test/resources/windows-created-offender-root.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/windows-created-offender-root.zip -------------------------------------------------------------------------------- /api/src/test/resources/windows-created-offender-traversal-elements.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/windows-created-offender-traversal-elements.zip -------------------------------------------------------------------------------- /api/src/test/resources/yolo_world/identity.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/api/src/test/resources/yolo_world/identity.pt -------------------------------------------------------------------------------- /api/src/test/resources/yolo_world/merges.txt: -------------------------------------------------------------------------------- 1 | #version: 0.2 2 | r e 3 | o n 4 | r o 5 | a t 6 | c on 7 | m o 8 | c o 9 | t r 10 | t e 11 | c a 12 | e m 13 | t ro 14 | o t 15 | con t 16 | c at 17 | con tro 18 | o l 19 | contro l 20 | re mo 21 | ro l 22 | remo te 23 | n t 24 | re m 25 | mo t 26 | on t -------------------------------------------------------------------------------- /api/src/test/resources/yolo_world/vocab.json: -------------------------------------------------------------------------------- 1 | { 2 | "control": 3366, 3 | "remote": 9687, 4 | "cat": 2368, 5 | "<|startoftext|>": 49406, 6 | "<|endoftext|>": 49407 7 | } -------------------------------------------------------------------------------- /apt.txt: -------------------------------------------------------------------------------- 1 | openjdk-17-jdk 2 | -------------------------------------------------------------------------------- /basicdataset/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /basicdataset/src/main/java/ai/djl/basicdataset/utils/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities used within the basic datasets. */ 15 | package ai.djl.basicdataset.utils; 16 | -------------------------------------------------------------------------------- /basicdataset/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) basic datasets.

7 | 8 |

9 | The basic datasets module contains a number of built-in datasets that can be used for deep learning. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /basicdataset/src/test/java/ai/djl/basicdataset/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the basicdataset module. */ 15 | package ai.djl.basicdataset; 16 | -------------------------------------------------------------------------------- /basicdataset/src/test/java/ai/djl/basicdataset/tabular/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2013 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.basicdataset.tabular}. */ 15 | package ai.djl.basicdataset.tabular; 16 | -------------------------------------------------------------------------------- /basicdataset/src/test/resources/imagefolder/cat/kitten.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/basicdataset/src/test/resources/imagefolder/cat/kitten.jpg -------------------------------------------------------------------------------- /basicdataset/src/test/resources/imagefolder/dog/dog_bike_car.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/basicdataset/src/test/resources/imagefolder/dog/dog_bike_car.jpg -------------------------------------------------------------------------------- /basicdataset/src/test/resources/imagefolder/misc/pikachu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/basicdataset/src/test/resources/imagefolder/misc/pikachu.png -------------------------------------------------------------------------------- /basicdataset/src/test/resources/mlrepo/dataset/cv/ai/djl/basicdataset/captcha/metadata.json: -------------------------------------------------------------------------------- 1 | { 2 | "metadataVersion": "0.2", 3 | "resourceType": "dataset", 4 | "application": "cv", 5 | "groupId": "ai.djl.basicdataset", 6 | "artifactId": "captcha", 7 | "name": "Captcha", 8 | "description": "A simple captcha dataset", 9 | "website": "http://www.djl.ai/basicdataset", 10 | "artifacts": [ 11 | { 12 | "version": "1.1", 13 | "snapshot": false, 14 | "files": { 15 | "data": { 16 | "uri": "captchaImage.zip", 17 | "sha1Hash": "575da8579b61de1b9d094f16cdfa0a8358394f45", 18 | "name": "data", 19 | "size": 43210912 20 | } 21 | } 22 | } 23 | ] 24 | } 25 | -------------------------------------------------------------------------------- /basicdataset/src/test/resources/mlrepo/dataset/nlp/ai/djl/basicdataset/stanford-movie-review/metadata.json: -------------------------------------------------------------------------------- 1 | { 2 | "metadataVersion": "0.2", 3 | "resourceType": "dataset", 4 | "application": "nlp", 5 | "groupId": "ai.djl.basicdataset", 6 | "artifactId": "stanford-movie-review", 7 | "name": "stanford-movie-review", 8 | "description": "A sentiment analysis dataset of IMDB movie reviews", 9 | "website": "https://ai.stanford.edu/~amaas/data/sentiment/", 10 | "artifacts": [ 11 | { 12 | "version": "1.0", 13 | "snapshot": false, 14 | "files": { 15 | "aclImdb": { 16 | "uri": "1.0/aclImdb.zip", 17 | "sha1Hash": "cd5f28ea7b6b33f68e1d8cf82938ba37eb929e39", 18 | "size": 116689188 19 | } 20 | } 21 | } 22 | ] 23 | } 24 | -------------------------------------------------------------------------------- /bom/gradle: -------------------------------------------------------------------------------- 1 | ../gradle -------------------------------------------------------------------------------- /bom/gradle.properties: -------------------------------------------------------------------------------- 1 | ../gradle.properties -------------------------------------------------------------------------------- /bom/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /bom/settings.gradle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/bom/settings.gradle -------------------------------------------------------------------------------- /build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | id("com.github.spotbugs") apply false 3 | ai.djl.release 4 | ai.djl.stats 5 | } 6 | -------------------------------------------------------------------------------- /buildSrc/build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | `kotlin-dsl` 3 | } 4 | 5 | repositories { 6 | mavenCentral() 7 | gradlePluginPortal() 8 | } 9 | 10 | dependencies { 11 | implementation("com.google.googlejavaformat:google-java-format:1.22.0") 12 | implementation("com.github.spotbugs.snom:spotbugs-gradle-plugin:6.0.15") 13 | implementation(files(libs.javaClass.superclass.protectionDomain.codeSource.location)) 14 | } 15 | -------------------------------------------------------------------------------- /buildSrc/settings.gradle.kts: -------------------------------------------------------------------------------- 1 | rootProject.name = "buildSrc" 2 | 3 | dependencyResolutionManagement { 4 | versionCatalogs { 5 | create("libs") { 6 | from(files("../gradle/libs.versions.toml")) 7 | } 8 | } 9 | } -------------------------------------------------------------------------------- /buildSrc/src/main/kotlin/ai/djl/javaBase.gradle.kts: -------------------------------------------------------------------------------- 1 | package ai.djl 2 | 3 | import org.gradle.accessors.dm.LibrariesForLibs 4 | import org.gradle.kotlin.dsl.java 5 | import org.gradle.kotlin.dsl.maven 6 | import org.gradle.kotlin.dsl.repositories 7 | import org.gradle.kotlin.dsl.the 8 | 9 | plugins { 10 | java 11 | } 12 | 13 | group = "ai.djl" 14 | val isRelease = project.hasProperty("release") || project.hasProperty("staging") 15 | val libs = the() 16 | version = libs.versions.djl.get() + if (isRelease) "" else "-SNAPSHOT" 17 | 18 | repositories { 19 | mavenCentral() 20 | maven("https://oss.sonatype.org/content/repositories/snapshots/") 21 | } 22 | -------------------------------------------------------------------------------- /djl-zero/build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | ai.djl.javaProject 3 | ai.djl.publish 4 | } 5 | 6 | dependencies { 7 | api(project(":api")) 8 | api(project(":basicdataset")) 9 | api(project(":model-zoo")) 10 | testImplementation(libs.slf4j.simple) 11 | 12 | testImplementation(project(":testing")) 13 | testImplementation(libs.testng) 14 | 15 | // Current engines and model zoos used for inference 16 | // runtimeOnly(project(":engines:pytorch:pytorch-engine")) 17 | // runtimeOnly(project(":engines:pytorch:pytorch-model-zoo")) 18 | runtimeOnly(project(":engines:mxnet:mxnet-engine")) 19 | runtimeOnly(project(":engines:mxnet:mxnet-model-zoo")) 20 | } 21 | -------------------------------------------------------------------------------- /djl-zero/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /djl-zero/src/main/java/ai/djl/zero/tabular/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains training for Tabular({@link ai.djl.Application.Tabular}) tasks. */ 15 | package ai.djl.zero.tabular; 16 | -------------------------------------------------------------------------------- /djl-zero/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) Zero API.

7 | 8 |

9 | The zero module contains a zero deep learning knowledge required wrapper over DJL. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /djl-zero/src/test/java/ai/djl/zero/cv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for {@link ai.djl.zero.cv}. */ 15 | package ai.djl.zero.cv; 16 | -------------------------------------------------------------------------------- /docker/windows/Dockerfile: -------------------------------------------------------------------------------- 1 | ARG version=ltsc2019 2 | FROM mcr.microsoft.com/windows/servercore:$version 3 | 4 | ADD https://aka.ms/vs/17/release/vc_redist.x64.exe /vc_redist.x64.exe 5 | RUN C:\vc_redist.x64.exe /quiet /install 6 | RUN del C:\vc_redist.x64.exe 7 | 8 | ENV chocolateyUseWindowsCompression false 9 | 10 | RUN powershell -Command \ 11 | Set-ExecutionPolicy Bypass -Scope Process -Force; [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072; iex ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1')); \ 12 | choco feature disable --name showDownloadProgress 13 | 14 | RUN choco install -y openjdk17 15 | -------------------------------------------------------------------------------- /docs/.gitignore: -------------------------------------------------------------------------------- 1 | demos 2 | serving 3 | -------------------------------------------------------------------------------- /docs/development/README.md: -------------------------------------------------------------------------------- 1 | # Contributor Documentation 2 | 3 | This folder contains documentation and guides for contributors and those working internally with the Deep Java Library (DJL) project. 4 | 5 | ## Main Guides 6 | 7 | - **[Setup development environment](setup.md)** 8 | - **[Development guideline](development_guideline.md)** 9 | - [Troubleshooting](troubleshooting.md) 10 | 11 | ## Model Zoo Guides 12 | 13 | - [How to add a new model to the model zoos](add_model_to_model-zoo.md) 14 | -------------------------------------------------------------------------------- /docs/development/img/custom_debug_view.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/docs/development/img/custom_debug_view.png -------------------------------------------------------------------------------- /docs/development/img/ndmanager_structure_for_inference.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/docs/development/img/ndmanager_structure_for_inference.png -------------------------------------------------------------------------------- /docs/development/img/ndmanager_structure_for_training.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/docs/development/img/ndmanager_structure_for_training.png -------------------------------------------------------------------------------- /docs/telemetry.md: -------------------------------------------------------------------------------- 1 | # Usage Tracking 2 | 3 | Since DJL 0.20.0, DJL collects telemetry to help us better understand our users’ needs 4 | when running on AWS EC2. DJL contains code that allows DJL development team to collect the 5 | AWS EC2 instance-type, instance-id and DJL version information. No other information about 6 | the system is collected or retained. 7 | 8 | To opt out of usage tracking for DJL, you can set the `OPT_OUT_TRACKING` environment variable: 9 | 10 | ```bash 11 | export OPT_OUT_TRACKING=true 12 | ``` 13 | 14 | or Java System property: 15 | 16 | ```java 17 | System.setProperty("OPT_OUT_TRACKING", "true") 18 | ``` 19 | 20 | Usage tracking is also disable in `offline` mode: 21 | 22 | ```java 23 | System.setProperty("ai.djl.offline", "true") 24 | ``` 25 | -------------------------------------------------------------------------------- /engines/ml/lightgbm/build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | ai.djl.javaProject 3 | ai.djl.publish 4 | } 5 | 6 | group = "ai.djl.ml.lightgbm" 7 | 8 | dependencies { 9 | api(project(":api")) 10 | api("com.microsoft.ml.lightgbm:lightgbmlib:${libs.versions.lightgbm.get()}") 11 | 12 | testImplementation(project(":testing")) 13 | 14 | testRuntimeOnly(libs.slf4j.simple) 15 | } 16 | 17 | publishing { 18 | publications { 19 | named("maven") { 20 | pom { 21 | name = "DJL Engine Adapter for LightGBM" 22 | description = "Deep Java Library (DJL) Engine Adapter for LightGBM" 23 | url = "https://djl.ai/engines/ml/${project.name}" 24 | } 25 | } 26 | } 27 | } 28 | -------------------------------------------------------------------------------- /engines/ml/lightgbm/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/ml/lightgbm/src/main/java/ai/djl/ml/lightgbm/jni/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains Helper class to access LightGBM JNI. */ 15 | package ai.djl.ml.lightgbm.jni; 16 | -------------------------------------------------------------------------------- /engines/ml/lightgbm/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) LightGBM Engine.

7 | 8 |

9 | The LightGBM Engine module contains the LightGBM implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/ml/lightgbm/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.ml.lightgbm.LgbmEngineProvider 2 | -------------------------------------------------------------------------------- /engines/ml/lightgbm/src/test/java/ai/djl/ml/lightgbm/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** The integration test for testing LightGBM specific features. */ 15 | package ai.djl.ml.lightgbm; 16 | -------------------------------------------------------------------------------- /engines/ml/xgboost/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/ml/xgboost/src/main/java/com/sun/jna/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains JNA changes on the existing Pointer. */ 15 | package com.sun.jna; 16 | -------------------------------------------------------------------------------- /engines/ml/xgboost/src/main/java/ml/dmlc/xgboost4j/java/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains Helper class to access XGBoost JNI. */ 15 | package ml.dmlc.xgboost4j.java; 16 | -------------------------------------------------------------------------------- /engines/ml/xgboost/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) XGBoost Engine.

7 | 8 |

9 | The XGBoost Engine module contains the XGBoost implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/ml/xgboost/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.ml.xgboost.XgbEngineProvider 2 | -------------------------------------------------------------------------------- /engines/ml/xgboost/src/test/java/ai/djl/ml/xgboost/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** The integration test for testing XGBoost specific features. */ 15 | package ai.djl.ml.xgboost; 16 | -------------------------------------------------------------------------------- /engines/mxnet/README.md: -------------------------------------------------------------------------------- 1 | # Apache MXNet Engine 2 | 3 | This directory contains the Deep Java Library (DJL) EngineProvider for [Apache MXNet (incubating) Deep Learning Framework](https://mxnet.apache.org/). 4 | 5 | ## Modules 6 | 7 | - [MXNet Engine](mxnet-engine/README.md) - The DJL implementation for Apache MXNet Engine 8 | - [MXNet Model Zoo](mxnet-model-zoo/README.md) - A ModelZoo containing models exported from Apache MXNet 9 | - [native](native/README.md) - A utility module for building the mxnet-native jars containing the native binaries 10 | - [jnarator](jnarator/README.md) - A utility module used by the MxEngine to call the Apache MXNet C API 11 | -------------------------------------------------------------------------------- /engines/mxnet/jnarator/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/mxnet/jnarator/src/main/resources/log4j2.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) EngineProvider for Apache MXNet (incubating) 7 | .

8 | 9 |

10 | The MXNet Engine module contains the DJL EngineProvider for Apache MXNet (incubating). 11 | See here for more details. 12 |

13 | 14 | 15 | 16 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/src/main/resources/META-INF/native-image/ai/djl/mxnet/mxnet-engine/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.mxnet.engine.MxEngineProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | } 9 | ] 10 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/src/main/resources/META-INF/native-image/ai/djl/mxnet/mxnet-engine/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.engine.EngineProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.mxnet.engine.MxEngineProvider 2 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-engine/src/main/resources/mxnet-engine.properties: -------------------------------------------------------------------------------- 1 | djl_version=${djlVersion} 2 | mxnet_version=${mxnetVersion} -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/main/java/ai/djl/mxnet/zoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the built-in {@link ai.djl.mxnet.zoo.MxModelZoo}. */ 15 | package ai.djl.mxnet.zoo; 16 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) Model Zoo provider for Apache MXNet (incubating).

7 | 8 |

9 | The MXNet Model Zoo module contains Apache MXNet (incubating) symbolic modules that can be used pre-trained or for inference. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/main/resources/META-INF/native-image/ai/djl/mxnet/mxnet-model-zoo/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.mxnet.zoo.MxZooProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | }, 9 | { 10 | "name": "ai.djl.mxnet.zoo.MxModelZoo", 11 | "allDeclaredFields": true 12 | } 13 | ] 14 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/main/resources/META-INF/native-image/ai/djl/mxnet/mxnet-model-zoo/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.repository.zoo.ZooProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.mxnet.zoo.MxZooProvider 2 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/test/resources/mlrepo/model/cv/image_classification/ai/djl/mxnet/mlp/synset.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | 8 10 | 9 -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/test/resources/mlrepo/model/cv/image_classification/ai/djl/mxnet/synset_cifar10.txt: -------------------------------------------------------------------------------- 1 | airplane 2 | automobile 3 | bird 4 | cat 5 | deer 6 | dog 7 | frog 8 | horse 9 | ship 10 | truck 11 | -------------------------------------------------------------------------------- /engines/mxnet/mxnet-model-zoo/src/test/resources/mlrepo/model/cv/object_detection/ai/djl/mxnet/classes_voc.txt: -------------------------------------------------------------------------------- 1 | aeroplane 2 | bicycle 3 | bird 4 | boat 5 | bottle 6 | bus 7 | car 8 | cat 9 | chair 10 | cow 11 | diningtable 12 | dog 13 | horse 14 | motorbike 15 | person 16 | pottedplant 17 | sheep 18 | sofa 19 | train 20 | tvmonitor -------------------------------------------------------------------------------- /engines/mxnet/native/.gitignore: -------------------------------------------------------------------------------- 1 | src/main/resources/binaries 2 | -------------------------------------------------------------------------------- /engines/mxnet/native/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/mxnet/native/src/main/resources/META-INF/backport.txt: -------------------------------------------------------------------------------- 1 | The file is based off Apache MXNet 1.7 branch (https://github.com/apache/incubator-mxnet/commit/7a84fe11dd7399d1c2bdd0fb679c46af69e48861) 2 | and has been patched with master commit (https://github.com/apache/incubator-mxnet/commit/5fd90249cafd14910633d25d3bc370a522736ac7). 3 | -------------------------------------------------------------------------------- /engines/mxnet/native/src/main/resources/META-INF/native-image/ai/djl/mxnet/mxnet-native/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "native/lib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-android/README.md: -------------------------------------------------------------------------------- 1 | # DJL - ONNX Runtime for Android 2 | 3 | ## Overview 4 | This module contains the DJL ONNX Runtime engine for Android. 5 | 6 | ## Installation 7 | You can pull the ONNX Runtime for Android from the central Maven repository by including the following dependency: 8 | 9 | - ai.djl.android:onnxruntime:0.33.0 10 | 11 | ```xml 12 | 13 | ai.djl.android 14 | onnxruntime 15 | 0.33.0 16 | runtime 17 | 18 | ``` 19 | -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-android/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-android/src/main/.empty: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/engines/onnxruntime/onnxruntime-android/src/main/.empty -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-engine/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-engine/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) ONNX Runtime Engine.

7 | 8 |

9 | The ONNX Runtime Engine module contains the ONNX Runtime implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-engine/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.onnxruntime.engine.OrtEngineProvider 2 | -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-engine/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.onnxruntime.zoo.OrtZooProvider 2 | ai.djl.onnxruntime.zoo.OrtHfZooProvider 3 | -------------------------------------------------------------------------------- /engines/onnxruntime/onnxruntime-engine/src/test/java/ai/djl/onnxruntime/zoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains test classes for onnxruntime model zoo. */ 14 | package ai.djl.onnxruntime.zoo; 15 | -------------------------------------------------------------------------------- /engines/pytorch/README.md: -------------------------------------------------------------------------------- 1 | # Pytorch Engine 2 | 3 | This directory contains the Deep Java Library (DJL) EngineProvider for PyTorch. 4 | 5 | It is based off the [PyTorch Deep Learning Framework](https://pytorch.org/). 6 | 7 | ## Modules 8 | 9 | - [PyTorch Engine](pytorch-engine/README.md) - The DJL implementation for PyTorch Engine 10 | - [PyTorch Model Zoo](pytorch-model-zoo/README.md) - A ModelZoo containing models exported from PyTorch 11 | - [Pytorch native library](pytorch-native/README.md) - A utility module for building the pytorch-native jars containing the native binaries 12 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) PyTorch Engine.

7 | 8 |

9 | The PyTorch Engine module contains the PyTorch implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-engine/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.pytorch.engine.PtEngineProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | } 9 | ] 10 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-engine/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.engine.EngineProvider" 5 | }, 6 | { 7 | "pattern": "jnilib/.*" 8 | } 9 | ] 10 | } 11 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.pytorch.engine.PtEngineProvider 2 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/main/resources/pytorch-engine.properties: -------------------------------------------------------------------------------- 1 | djl_version=${djlVersion} 2 | pytorch_version=${pytorchVersion} -------------------------------------------------------------------------------- /engines/pytorch/pytorch-engine/src/test/java/ai/djl/pytorch/jni/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** The integration test for testing PyTorch specific features. */ 14 | package ai.djl.pytorch.jni; 15 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-jni/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/pytorch/pytorch-jni/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-jni/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "jnilib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-model-zoo/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/pytorch/pytorch-model-zoo/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) PyTorch Model Zoo.

7 | 8 |

9 | The PyTorch Model Zoo module contains PyTorch pretrained traced modules that can be used for inference. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-model-zoo/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-model-zoo/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.pytorch.zoo.PtZooProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | }, 9 | { 10 | "name": "ai.djl.pytorch.zoo.PtModelZoo", 11 | "allDeclaredFields": true 12 | } 13 | ] 14 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-model-zoo/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-model-zoo/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.repository.zoo.ZooProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-model-zoo/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.pytorch.zoo.PtZooProvider 2 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-native/.gitignore: -------------------------------------------------------------------------------- 1 | cmake-build-*/ 2 | libtorch 3 | libtorch_android 4 | jnilib/ 5 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-native/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/pytorch/pytorch-native/src/main/java/ai/djl/pytorch/jni/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains helper class to load native shared library. */ 14 | package ai.djl.pytorch.jni; 15 | -------------------------------------------------------------------------------- /engines/pytorch/pytorch-native/src/main/resources/META-INF/native-image/ai/djl/pytorch/pytorch-native/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "native/lib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-api/.gitignore: -------------------------------------------------------------------------------- 1 | libs/ 2 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-api/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-api/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-api/native-image.properties: -------------------------------------------------------------------------------- 1 | Args = -H:ReflectionConfigurationResources=${.}/reflect-config.json \ 2 | --initialize-at-run-time=org.tensorflow.EagerSession 3 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) Tensorflow Engine.

7 | 8 |

9 | The Tensorflow Engine module contains the Tensorflow implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-engine/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.tensorflow.engine.TfEngineProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | } 9 | ] 10 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-engine/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.engine.EngineProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.tensorflow.engine.TfEngineProvider 2 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/main/resources/tensorflow-engine.properties: -------------------------------------------------------------------------------- 1 | djl_version=${djlVersion} 2 | tensorflow_version=${tensorflowVersion} -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-engine/src/test/java/ai/djl/tensorflow/engine/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** The test classes for TensorFlow engine features. */ 14 | package ai.djl.tensorflow.engine; 15 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-model-zoo/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-model-zoo/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) TensorFlow Model Zoo.

7 | 8 |

9 | The TensorFlow Model Zoo module contains TensorFlow pretrained SavedModel modules that can be used for inference. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-model-zoo/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-model-zoo/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.tensorflow.zoo.TfZooProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | }, 9 | { 10 | "name": "ai.djl.tensorflow.zoo.TfModelZoo", 11 | "allDeclaredFields": true 12 | } 13 | ] 14 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-model-zoo/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-model-zoo/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.repository.zoo.ZooProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-model-zoo/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.tensorflow.zoo.TfZooProvider 2 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-native/README.md: -------------------------------------------------------------------------------- 1 | # DJL - TensorFlow native library 2 | 3 | This module publish a placeholder native library that will detect your current platform 4 | and download native TensorFlow library at runtime. 5 | -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-native/gradlew: -------------------------------------------------------------------------------- 1 | ../../../gradlew -------------------------------------------------------------------------------- /engines/tensorflow/tensorflow-native/src/main/resources/META-INF/native-image/ai/djl/tensorflow/tensorflow-native/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "native/lib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /engines/tensorrt/.gitignore: -------------------------------------------------------------------------------- 1 | /trt 2 | /cmake-build-debug 3 | /jnilib 4 | -------------------------------------------------------------------------------- /engines/tensorrt/build.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | set -e 4 | WORK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" 5 | export WORK_DIR 6 | 7 | VERSION="$(awk -F '=' '/tensorrt/ {gsub(/ ?"/, "", $2); print $2}' ../../gradle/libs.versions.toml)" 8 | 9 | if [ ! -d "trt" ]; 10 | then 11 | git clone https://github.com/NVIDIA/TensorRT.git -b $VERSION trt 12 | fi 13 | 14 | if [ ! -d "build" ]; 15 | then 16 | mkdir build 17 | fi 18 | cd build 19 | if [ ! -d "classes" ]; 20 | then 21 | mkdir classes 22 | fi 23 | javac -sourcepath ../src/main/java/ ../src/main/java/ai/djl/tensorrt/jni/TrtLibrary.java -h include -d classes 24 | cmake .. 25 | cmake --build . --config Release -- -j "$(nproc)" 26 | -------------------------------------------------------------------------------- /engines/tensorrt/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /engines/tensorrt/src/main/java/ai/djl/tensorrt/engine/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to interface with the underlying TensorRT Engine. */ 15 | package ai.djl.tensorrt.engine; 16 | -------------------------------------------------------------------------------- /engines/tensorrt/src/main/java/ai/djl/tensorrt/jni/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to interface with the underlying TensorRT Engine. */ 15 | package ai.djl.tensorrt.jni; 16 | -------------------------------------------------------------------------------- /engines/tensorrt/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) TensorRT Engine.

7 | 8 |

9 | The TensorRT Engine module contains the TensorRT implementation of the DJL EngineProvider. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /engines/tensorrt/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.tensorrt.engine.TrtEngineProvider 2 | -------------------------------------------------------------------------------- /engines/tensorrt/src/main/resources/native/lib/tensorrt.properties: -------------------------------------------------------------------------------- 1 | version=${trtVersion}-${version} -------------------------------------------------------------------------------- /engines/tensorrt/src/test/java/ai/djl/tensorrt/engine/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to interface with the underlying TensorRT Engine. */ 15 | package ai.djl.tensorrt.engine; 16 | -------------------------------------------------------------------------------- /engines/tensorrt/src/test/resources/identity.onnx: -------------------------------------------------------------------------------- 1 |  backend-test:[ 2 |  3 | xy"Identity test_identityZ 4 | x 5 |  6 |  7 |  8 |  9 | b 10 | y 11 |  12 |  13 |  14 |  15 | B -------------------------------------------------------------------------------- /engines/tensorrt/src/test/resources/identity_70.trt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/engines/tensorrt/src/test/resources/identity_70.trt -------------------------------------------------------------------------------- /engines/tensorrt/src/test/resources/identity_75.trt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/engines/tensorrt/src/test/resources/identity_75.trt -------------------------------------------------------------------------------- /examples/.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | .idea 3 | *.iml 4 | build 5 | target 6 | -------------------------------------------------------------------------------- /examples/.mvn/wrapper/.gitignore: -------------------------------------------------------------------------------- 1 | maven-wrapper.jar 2 | 3 | -------------------------------------------------------------------------------- /examples/.mvn/wrapper/maven-wrapper.properties: -------------------------------------------------------------------------------- 1 | distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.6.3/apache-maven-3.6.3-bin.zip 2 | wrapperUrl=https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar 3 | -------------------------------------------------------------------------------- /examples/docs/img/cat_dog_detected.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/cat_dog_detected.jpg -------------------------------------------------------------------------------- /examples/docs/img/detected-dog_bike_car.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/detected-dog_bike_car.png -------------------------------------------------------------------------------- /examples/docs/img/detected-dogs.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/detected-dogs.jpg -------------------------------------------------------------------------------- /examples/docs/img/detected_instances.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/detected_instances.png -------------------------------------------------------------------------------- /examples/docs/img/retinaface_detected.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/retinaface_detected.png -------------------------------------------------------------------------------- /examples/docs/img/workFlow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/docs/img/workFlow.png -------------------------------------------------------------------------------- /examples/gradle: -------------------------------------------------------------------------------- 1 | ../gradle -------------------------------------------------------------------------------- /examples/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/clip/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of Text and Image vectorization. */ 15 | package ai.djl.examples.inference.clip; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/cv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains examples of CV. */ 14 | package ai.djl.examples.inference.cv; 15 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/face/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of face detection. */ 15 | package ai.djl.examples.inference.face; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/nlp/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of NLP. */ 15 | package ai.djl.examples.inference.nlp; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of performing inference using pre-trained models. */ 15 | package ai.djl.examples.inference; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/sr/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of image enhancements. */ 15 | package ai.djl.examples.inference.sr; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/timeseries/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of time series forecasting. */ 15 | package ai.djl.examples.inference.timeseries; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/inference/whisper/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of Speech to Text task. */ 15 | package ai.djl.examples.inference.whisper; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/training/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of training models. */ 15 | package ai.djl.examples.training; 16 | -------------------------------------------------------------------------------- /examples/src/main/java/ai/djl/examples/training/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities used for the training examples. */ 15 | package ai.djl.examples.training.util; 16 | -------------------------------------------------------------------------------- /examples/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) examples.

7 | 8 |

9 | The examples module contains a number of examples of training and inference using DJL. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /examples/src/main/resources/log4j2.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/clip/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains examples of Text and Image vectorization. */ 15 | package ai.djl.examples.inference.clip; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/cv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference.cv; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/face/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference.face; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/nlp/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference.nlp; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/sr/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference.sr; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/inference/timeseries/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the inference examples. */ 15 | package ai.djl.examples.inference.timeseries; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/examples/training/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the training examples. */ 15 | package ai.djl.examples.training; 16 | -------------------------------------------------------------------------------- /examples/src/test/java/ai/djl/testing/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains testing utilities for other modules. */ 15 | package ai.djl.testing; 16 | -------------------------------------------------------------------------------- /examples/src/test/resources/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/0.png -------------------------------------------------------------------------------- /examples/src/test/resources/action_discus_throw.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/action_discus_throw.png -------------------------------------------------------------------------------- /examples/src/test/resources/airplane1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/airplane1.png -------------------------------------------------------------------------------- /examples/src/test/resources/dog-cat.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/dog-cat.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/dog_bike_car.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/dog_bike_car.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/fox.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/fox.png -------------------------------------------------------------------------------- /examples/src/test/resources/kana1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/kana1.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/kana2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/kana2.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/kitten.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/kitten.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/largest_selfie.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/largest_selfie.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/mountains.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/mountains.png -------------------------------------------------------------------------------- /examples/src/test/resources/pikachu.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/pikachu.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/pose_soccer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/pose_soccer.png -------------------------------------------------------------------------------- /examples/src/test/resources/segmentation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/segmentation.jpg -------------------------------------------------------------------------------- /examples/src/test/resources/yolov8_test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/examples/src/test/resources/yolov8_test.jpg -------------------------------------------------------------------------------- /extensions/audio/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/audio/src/main/java/ai/djl/audio/dataset/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains useful audio datasets. */ 15 | package ai.djl.audio.dataset; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/main/java/ai/djl/audio/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains audio utils and datasets for djl. */ 15 | package ai.djl.audio; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/main/java/ai/djl/audio/processor/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains multiple processors for audio and signal processing. */ 15 | package ai.djl.audio.processor; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/main/java/ai/djl/audio/translator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains translators for audio processing. */ 15 | package ai.djl.audio.translator; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/main/java/ai/djl/audio/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains audio utils for djl. */ 15 | package ai.djl.audio.util; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Audio extension for DJL.

7 | 8 |

The audio module contains classes that provide high performance audio processing for DJL.

9 | 10 | 11 | -------------------------------------------------------------------------------- /extensions/audio/src/test/java/ai/djl/audio/dataset/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the dataset module. */ 15 | package ai.djl.audio.dataset; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/test/java/ai/djl/audio/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the audio extension module. */ 15 | package ai.djl.audio; 16 | -------------------------------------------------------------------------------- /extensions/audio/src/test/java/ai/djl/audio/processor/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the processor module. */ 15 | package ai.djl.audio.processor; 16 | -------------------------------------------------------------------------------- /extensions/aws-ai/build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | ai.djl.javaProject 3 | ai.djl.publish 4 | } 5 | 6 | group = "ai.djl.aws" 7 | 8 | dependencies { 9 | api(platform(libs.awssdk.bom)) 10 | api(libs.awssdk.s3) 11 | api(project(":api")) 12 | 13 | testImplementation(project(":engines:pytorch:pytorch-model-zoo")) 14 | 15 | testImplementation(libs.testng) 16 | testImplementation(libs.apache.log4j.slf4j) 17 | } 18 | 19 | publishing { 20 | publications { 21 | named("maven") { 22 | pom { 23 | name = "AWS AI toolkit for DJL" 24 | description = "AWS AI toolkit for DJL" 25 | url = "http://www.djl.ai/extensions/${project.name}" 26 | } 27 | } 28 | } 29 | } 30 | -------------------------------------------------------------------------------- /extensions/aws-ai/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/aws-ai/src/main/resources/META-INF/services/ai.djl.repository.RepositoryFactory: -------------------------------------------------------------------------------- 1 | ai.djl.aws.s3.S3RepositoryFactory 2 | -------------------------------------------------------------------------------- /extensions/aws-ai/src/main/resources/ai/djl/aws/sagemaker/assume_role_policy.json: -------------------------------------------------------------------------------- 1 | { 2 | "Version": "2012-10-17", 3 | "Statement": [ 4 | { 5 | "Effect": "Allow", 6 | "Principal": { 7 | "Service": "sagemaker.amazonaws.com" 8 | }, 9 | "Action": "sts:AssumeRole" 10 | } 11 | ] 12 | } 13 | -------------------------------------------------------------------------------- /extensions/aws-ai/src/main/resources/ai/djl/aws/sagemaker/execution_policy.json: -------------------------------------------------------------------------------- 1 | { 2 | "Version": "2012-10-17", 3 | "Statement": [ 4 | { 5 | "Sid": "VisualEditor0", 6 | "Effect": "Allow", 7 | "Action": "s3:ListBucket", 8 | "Resource": [ 9 | "arn:aws:s3:::*" 10 | ] 11 | }, 12 | { 13 | "Sid": "VisualEditor1", 14 | "Effect": "Allow", 15 | "Action": [ 16 | "s3:PutObject", 17 | "s3:GetObject", 18 | "s3:DeleteObject" 19 | ], 20 | "Resource": [ 21 | "arn:aws:s3:::*" 22 | ] 23 | } 24 | ] 25 | } 26 | -------------------------------------------------------------------------------- /extensions/aws-ai/src/test/java/ai/djl/aws/s3/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the AWS s3 module. */ 15 | package ai.djl.aws.s3; 16 | -------------------------------------------------------------------------------- /extensions/aws-ai/src/test/resources/log4j2.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | 21 | -------------------------------------------------------------------------------- /extensions/benchmark/README.md: -------------------------------------------------------------------------------- 1 | # Benchmark your DL model 2 | 3 | dll-bench is a command line tool that makes it easy for you to benchmark the model on different 4 | platforms. 5 | 6 | With djl-bench, you can easily compare your model's behavior in different use cases, such as: 7 | 8 | - single-threaded vs. multi-threaded 9 | - single input vs. batched inputs 10 | - CPU vs. GPU or other hardware accelerator 11 | - Single GPU vs. multiple GPUs 12 | - default engine options vs. customized engine configuration 13 | - running with different engines 14 | - running with different version of the engine 15 | 16 | 17 | **This module has been moved to [deepjavalibrary/djl-serving/benchmark](https://github.com/deepjavalibrary/djl-serving/tree/master/benchmark).** 18 | -------------------------------------------------------------------------------- /extensions/fasttext/.gitignore: -------------------------------------------------------------------------------- 1 | /fastText 2 | /cmake-build-debug 3 | -------------------------------------------------------------------------------- /extensions/fasttext/CMakeLists.txt: -------------------------------------------------------------------------------- 1 | cmake_minimum_required(VERSION 3.6 FATAL_ERROR) 2 | project(jni_fasttext) 3 | 4 | set(CMAKE_CXX_STANDARD 11) 5 | set(CMAKE_CXX_STANDARD_REQUIRED ON) 6 | 7 | set(JAVA_AWT_LIBRARY NotNeeded) 8 | set(JAVA_AWT_INCLUDE_PATH NotNeeded) 9 | find_package(JNI REQUIRED) 10 | find_path(UTILS_INCLUDE_DIR NAMES djl/utils.h PATHS ${PROJECT_SOURCE_DIR}/../../api/src/main/native REQUIRED) 11 | add_subdirectory(fastText) 12 | 13 | add_library(jni_fasttext SHARED src/main/native/ai_djl_fasttext_jni_FastTextLibrary.cc) 14 | target_include_directories(jni_fasttext PUBLIC 15 | ${JNI_INCLUDE_DIRS} 16 | ${UTILS_INCLUDE_DIR} 17 | fastText/src 18 | build/include) 19 | target_link_libraries(jni_fasttext fasttext-static_pic) 20 | -------------------------------------------------------------------------------- /extensions/fasttext/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/fasttext/src/main/java/ai/djl/fasttext/jni/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to interface with the underlying fastText. */ 15 | package ai.djl.fasttext.jni; 16 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/java/ai/djl/fasttext/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the main fastText implementation of the DJL API. */ 15 | package ai.djl.fasttext; 16 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/java/ai/djl/fasttext/zoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the built-in {@link ai.djl.fasttext.zoo.FtModelZoo}. */ 15 | package ai.djl.fasttext.zoo; 16 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) NLP support with fastText.

7 | 8 |

9 | The fastText module contains the NLP suport with fastText. 10 |

11 | 12 | 13 | 14 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/resources/META-INF/native-image/ai/djl/fasttext/fasttext/jni-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "java.lang.NullPointerException" 4 | }, 5 | { 6 | "name": "java.lang.IllegalStateException" 7 | }, 8 | { 9 | "name": "java.util.HashSet", 10 | "allDeclaredConstructors": true, 11 | "allPublicConstructors": true, 12 | "allDeclaredMethods": true, 13 | "allPublicMethods": true 14 | }, 15 | { 16 | "name": "org.slf4j.Logger", 17 | "allDeclaredConstructors": true, 18 | "allPublicConstructors": true, 19 | "allDeclaredMethods": true, 20 | "allPublicMethods": true 21 | } 22 | ] 23 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/resources/META-INF/native-image/ai/djl/fasttext/fasttext/native-image.properties: -------------------------------------------------------------------------------- 1 | Args = -H:JNIConfigurationResources=${.}/jni-config.json \ 2 | -H:ResourceConfigurationResources=${.}/resource-config.json \ 3 | --initialize-at-run-time=ai.djl.fasttext.FtWrapper 4 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/resources/META-INF/native-image/ai/djl/fasttext/fasttext/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "native/lib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.fasttext.zoo.FtZooProvider 2 | -------------------------------------------------------------------------------- /extensions/fasttext/src/main/resources/native/lib/fasttext.properties: -------------------------------------------------------------------------------- 1 | version=${fasttextVersion}-${version} -------------------------------------------------------------------------------- /extensions/fasttext/src/test/java/ai/djl/fasttext/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.fasttext}. */ 14 | package ai.djl.fasttext; 15 | -------------------------------------------------------------------------------- /extensions/hadoop/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/hadoop/src/main/resources/META-INF/services/ai.djl.repository.RepositoryFactory: -------------------------------------------------------------------------------- 1 | ai.djl.hadoop.hdfs.HdfsRepositoryFactory 2 | -------------------------------------------------------------------------------- /extensions/hadoop/src/test/java/ai/djl/hadoop/hdfs/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the hadoop module. */ 15 | package ai.djl.hadoop.hdfs; 16 | -------------------------------------------------------------------------------- /extensions/opencv/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/opencv/src/main/java/ai/djl/opencv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains classes that provides high performance image processing functionalities. */ 14 | package ai.djl.opencv; 15 | -------------------------------------------------------------------------------- /extensions/opencv/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the OpenCV for DJL.

7 | 8 |

The opencv module contains classes that provides high performance image processing for DJL.

9 | 10 | 11 | -------------------------------------------------------------------------------- /extensions/opencv/src/test/java/ai/djl/opencv/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2021 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for the opencv module. */ 14 | package ai.djl.opencv; 15 | -------------------------------------------------------------------------------- /extensions/sentencepiece/.gitignore: -------------------------------------------------------------------------------- 1 | /sentencepiece 2 | /cmake-build-debug 3 | -------------------------------------------------------------------------------- /extensions/sentencepiece/build.cmd: -------------------------------------------------------------------------------- 1 | set VERSION="%1" 2 | set FILEPATH="sentencepiece" 3 | 4 | if exist %FILEPATH% ( 5 | echo Found %FILEPATH% 6 | ) else ( 7 | echo Couldn't find %FILEPATH%, downloading it ... 8 | echo Cloning google/sentencepiece.git 9 | git clone https://github.com/google/sentencepiece.git -b %VERSION% 10 | echo Finished cloning sentencepiece 11 | ) 12 | 13 | if exist build rd /q /s build 14 | md build\classes 15 | cd build 16 | 17 | javac -sourcepath ../src/main/java/ ../src/main/java/ai/djl/sentencepiece/jni/SentencePieceLibrary.java -h include -d classes 18 | cmake .. 19 | cmake --build . --config Release 20 | 21 | :: for nightly ci 22 | md jnilib\win-x86_64 23 | copy Release\sentencepiece_native.dll jnilib\win-x86_64 24 | 25 | cd .. 26 | -------------------------------------------------------------------------------- /extensions/sentencepiece/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/sentencepiece/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) EngineProvider for SentencePiece.

7 | 8 |

9 | The SentencePiece Engine module contains the DJL EngineProvider for SentencePiece. 10 |

11 | 12 | 13 | 14 | -------------------------------------------------------------------------------- /extensions/sentencepiece/src/main/resources/META-INF/native-image/ai/djl/sentencepiece/sentencepiece/jni-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "java.lang.NullPointerException" 4 | }, 5 | { 6 | "name": "java.lang.IllegalStateException" 7 | }, 8 | { 9 | "name": "java.util.HashSet", 10 | "allDeclaredConstructors": true, 11 | "allPublicConstructors": true, 12 | "allDeclaredMethods": true, 13 | "allPublicMethods": true 14 | }, 15 | { 16 | "name": "org.slf4j.Logger", 17 | "allDeclaredConstructors": true, 18 | "allPublicConstructors": true, 19 | "allDeclaredMethods": true, 20 | "allPublicMethods": true 21 | } 22 | ] 23 | -------------------------------------------------------------------------------- /extensions/sentencepiece/src/main/resources/META-INF/native-image/ai/djl/sentencepiece/sentencepiece/native-image.properties: -------------------------------------------------------------------------------- 1 | Args = -H:JNIConfigurationResources=${.}/jni-config.json \ 2 | -H:ResourceConfigurationResources=${.}/resource-config.json \ 3 | --initialize-at-run-time=ai.djl.sentencepiece.SpProcessor 4 | -------------------------------------------------------------------------------- /extensions/sentencepiece/src/main/resources/META-INF/native-image/ai/djl/sentencepiece/sentencepiece/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "native/lib/.*" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /extensions/sentencepiece/src/main/resources/native/lib/sentencepiece.properties: -------------------------------------------------------------------------------- 1 | version=${sentencepieceVersion}-${version} -------------------------------------------------------------------------------- /extensions/sentencepiece/src/test/java/ai/djl/sentencepiece/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.sentencepiece}. */ 14 | package ai.djl.sentencepiece; 15 | -------------------------------------------------------------------------------- /extensions/spark/.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | dist/ 3 | *.egg-info/ 4 | *.egg 5 | __pycache__/ -------------------------------------------------------------------------------- /extensions/spark/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/spark/setup/PyPiDescription.rst: -------------------------------------------------------------------------------- 1 | Project Description 2 | =================== 3 | 4 | DJL Spark extension provides a set of APIs to make it easy to use DJL with Spark. It provides a set of APIs to load 5 | data from Spark DataFrames and run inference. 6 | -------------------------------------------------------------------------------- /extensions/spark/setup/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # 3 | # Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 4 | # 5 | # Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file 6 | # except in compliance with the License. A copy of the License is located at 7 | # 8 | # http://aws.amazon.com/apache2.0/ 9 | # 10 | # or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" 11 | # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for 12 | # the specific language governing permissions and limitations under the License. -------------------------------------------------------------------------------- /extensions/spark/setup/djl_spark/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # 3 | # Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 4 | # 5 | # Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file 6 | # except in compliance with the License. A copy of the License is located at 7 | # 8 | # http://aws.amazon.com/apache2.0/ 9 | # 10 | # or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" 11 | # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for 12 | # the specific language governing permissions and limitations under the License. 13 | -------------------------------------------------------------------------------- /extensions/spark/setup/djl_spark/task/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # 3 | # Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 4 | # 5 | # Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file 6 | # except in compliance with the License. A copy of the License is located at 7 | # 8 | # http://aws.amazon.com/apache2.0/ 9 | # 10 | # or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" 11 | # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for 12 | # the specific language governing permissions and limitations under the License. -------------------------------------------------------------------------------- /extensions/spark/setup/djl_spark/util/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # 3 | # Copyright 2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. 4 | # 5 | # Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file 6 | # except in compliance with the License. A copy of the License is located at 7 | # 8 | # http://aws.amazon.com/apache2.0/ 9 | # 10 | # or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" 11 | # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for 12 | # the specific language governing permissions and limitations under the License. -------------------------------------------------------------------------------- /extensions/tablesaw/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/tablesaw/src/main/java/ai/djl/tablesaw/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to support tabular datasets in Tablesaw format. */ 15 | package ai.djl.tablesaw; 16 | -------------------------------------------------------------------------------- /extensions/tablesaw/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Tablesaw for DJL.

7 | 8 |

The tablesaw module contains classes for DJL to support tabular datasets in Tablesaw format.

9 | 10 | 11 | -------------------------------------------------------------------------------- /extensions/tablesaw/src/test/java/ai/djl/tablesaw/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the tablesaw module. */ 15 | package ai.djl.tablesaw; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/block/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the basic block classes. */ 15 | package ai.djl.timeseries.block; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/dataset/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the basic dataset classes. */ 15 | package ai.djl.timeseries.dataset; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/evaluator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the evaluator classes. */ 15 | package ai.djl.timeseries.evaluator; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/model/deepar/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains blocks for deepar models. */ 15 | package ai.djl.timeseries.model.deepar; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to support TimeSeries in djl. */ 15 | package ai.djl.timeseries; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/main/java/ai/djl/timeseries/transform/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains transform classes. */ 15 | package ai.djl.timeseries.transform; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/test/java/ai/djl/timeseries/block/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the basic block module. */ 15 | package ai.djl.timeseries.block; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/test/java/ai/djl/timeseries/dataset/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the dataset module. */ 15 | package ai.djl.timeseries.dataset; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/test/java/ai/djl/timeseries/evaluator/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the evaluator module. */ 15 | package ai.djl.timeseries.evaluator; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/test/java/ai/djl/timeseries/model/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the model module. */ 15 | package ai.djl.timeseries.model; 16 | -------------------------------------------------------------------------------- /extensions/timeseries/src/test/java/ai/djl/timeseries/timefeature/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the timeFeature module. */ 15 | package ai.djl.timeseries.timefeature; 16 | -------------------------------------------------------------------------------- /extensions/tokenizers/.gitignore: -------------------------------------------------------------------------------- 1 | /tokenizers 2 | /jnilib 3 | dist/ 4 | model/ 5 | tmp/ 6 | models.json 7 | -------------------------------------------------------------------------------- /extensions/tokenizers/build.cmd: -------------------------------------------------------------------------------- 1 | @rem https://chocolatey.org/docs/installation#install-with-cmdexe 2 | @rem to install rust java etc.. 3 | @rem choco install rust -y 4 | @rem choco install jdk8 -y 5 | 6 | if exist build rd /q /s build 7 | md build\classes 8 | 9 | set RUST_MANIFEST=rust/Cargo.toml 10 | cargo build --manifest-path %RUST_MANIFEST% --release 11 | 12 | @rem for nightly ci 13 | md build\jnilib\win-x86_64\cpu 14 | copy rust\target\release\djl_tokenizer.dll build\jnilib\win-x86_64\cpu\tokenizers.dll 15 | -------------------------------------------------------------------------------- /extensions/tokenizers/build_android.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | set -ex 4 | 5 | WORK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" 6 | export WORK_DIR 7 | 8 | FLAVOR=$1 9 | 10 | if [ ! -d "build" ]; then 11 | mkdir build 12 | fi 13 | 14 | pushd . 15 | 16 | if [ ! -d "build" ]; then 17 | mkdir build 18 | fi 19 | 20 | rm -rf build/classes 21 | mkdir build/classes 22 | 23 | javac -sourcepath src/main/java/ src/main/java/ai/djl/huggingface/tokenizers/jni/TokenizersLibrary.java -h build/include -d build/classes 24 | javac -sourcepath src/main/java/ src/main/java/ai/djl/engine/rust/RustLibrary.java -h build/include -d build/classes 25 | 26 | cd rust/ 27 | cargo ndk -t $FLAVOR -o $WORK_DIR/build/jnilib --platform=21 build --release 28 | cd .. 29 | popd 30 | -------------------------------------------------------------------------------- /extensions/tokenizers/gradlew: -------------------------------------------------------------------------------- 1 | ../../gradlew -------------------------------------------------------------------------------- /extensions/tokenizers/rust/.gitignore: -------------------------------------------------------------------------------- 1 | target 2 | -------------------------------------------------------------------------------- /extensions/tokenizers/rust/src/layers/mod.rs: -------------------------------------------------------------------------------- 1 | #[allow(dead_code, unused)] 2 | mod cublaslt; 3 | mod layer_norm; 4 | mod linear; 5 | #[allow(dead_code, unused)] 6 | mod rms_norm; 7 | 8 | pub use layer_norm::LayerNorm; 9 | pub use linear::{HiddenAct, Linear}; 10 | pub use rms_norm::RmsNorm; 11 | -------------------------------------------------------------------------------- /extensions/tokenizers/rust/src/layers/rms_norm.rs: -------------------------------------------------------------------------------- 1 | use candle::{DType, Device, Module, Result, Tensor, D}; 2 | use candle_nn::VarBuilder; 3 | 4 | #[derive(Debug, Clone)] 5 | pub struct RmsNorm { 6 | inner: candle_nn::RmsNorm, 7 | span: tracing::Span, 8 | } 9 | 10 | impl RmsNorm { 11 | pub fn load(vb: VarBuilder, size: usize, eps: f64) -> Result { 12 | let span = tracing::span!(tracing::Level::TRACE, "rms-norm"); 13 | let inner = candle_nn::rms_norm(size, eps, vb)?; 14 | Ok(Self { inner, span }) 15 | } 16 | } 17 | 18 | impl Module for RmsNorm { 19 | fn forward(&self, x: &Tensor) -> Result { 20 | let _enter = self.span.enter(); 21 | self.inner.forward(x) 22 | } 23 | } 24 | -------------------------------------------------------------------------------- /extensions/tokenizers/rust/src/utils.rs: -------------------------------------------------------------------------------- 1 | use candle::{Result, Tensor}; 2 | 3 | /// Repeats a key or value tensor for grouped query attention 4 | /// The input tensor should have a shape `(batch, num_kv_heads, seq_len, head_dim)`, 5 | pub fn repeat_kv(xs: Tensor, n_rep: usize) -> Result { 6 | if n_rep == 1 { 7 | Ok(xs) 8 | } else { 9 | let (b_sz, n_kv_head, seq_len, head_dim) = xs.dims4()?; 10 | // Using cat is faster than a broadcast as it avoids going through a potentially 11 | // strided copy. 12 | // https://github.com/huggingface/candle/pull/2043 13 | Tensor::cat(&vec![&xs; n_rep], 2)?.reshape((b_sz, n_kv_head * n_rep, seq_len, head_dim)) 14 | } 15 | } 16 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/java/ai/djl/engine/rust/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains classes to interface with the underlying native engine. */ 15 | package ai.djl.engine.rust; 16 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) Huggingface tokenizers extension.

7 | 8 |

9 | The Tokeniszers module contains the DJL extension for Huggingface tokenizers. 10 |

11 | 12 | 13 | 14 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/python/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | *.egg-info/ 3 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/python/djl_converter/__init__.py: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env python 2 | # 3 | # Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 4 | # 5 | # Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file 6 | # except in compliance with the License. A copy of the License is located at 7 | # 8 | # http://aws.amazon.com/apache2.0/ 9 | # 10 | # or in the "LICENSE.txt" file accompanying this file. This file is distributed on an "AS IS" 11 | # BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, express or implied. See the License for 12 | # the specific language governing permissions and limitations under the License. 13 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/python/requirements.txt: -------------------------------------------------------------------------------- 1 | huggingface-hub 2 | transformers 3 | sentence-transformers 4 | torch 5 | protobuf==3.20.2 6 | optimum[exporters] 7 | safetensors 8 | requests -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/resources/META-INF/services/ai.djl.engine.EngineProvider: -------------------------------------------------------------------------------- 1 | ai.djl.engine.rust.RsEngineProvider 2 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.huggingface.zoo.HfZooProvider 2 | ai.djl.engine.rust.zoo.RsZooProvider 3 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/main/resources/native/lib/tokenizers.properties: -------------------------------------------------------------------------------- 1 | version=${tokenizersVersion}-${version} -------------------------------------------------------------------------------- /extensions/tokenizers/src/test/java/ai/djl/engine/rust/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the rust module. */ 15 | package ai.djl.engine.rust; 16 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/test/java/ai/djl/engine/rust/zoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2024 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the rust zoo module. */ 15 | package ai.djl.engine.rust.zoo; 16 | -------------------------------------------------------------------------------- /extensions/tokenizers/src/test/java/ai/djl/huggingface/zoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2022 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests for the huggingface zoo module. */ 15 | package ai.djl.huggingface.zoo; 16 | -------------------------------------------------------------------------------- /gradle.properties: -------------------------------------------------------------------------------- 1 | org.gradle.daemon=true 2 | org.gradle.jvmargs=-Xmx1024M --add-exports jdk.compiler/com.sun.tools.javac.api=ALL-UNNAMED \ 3 | --add-exports jdk.compiler/com.sun.tools.javac.file=ALL-UNNAMED \ 4 | --add-exports jdk.compiler/com.sun.tools.javac.parser=ALL-UNNAMED \ 5 | --add-exports jdk.compiler/com.sun.tools.javac.tree=ALL-UNNAMED \ 6 | --add-exports jdk.compiler/com.sun.tools.javac.util=ALL-UNNAMED 7 | 8 | systemProp.org.gradle.internal.http.socketTimeout=120000 9 | systemProp.org.gradle.internal.http.connectionTimeout=60000 10 | 11 | # FIXME: Workaround gradle publish issue: https://github.com/gradle/gradle/issues/11308 12 | systemProp.org.gradle.internal.publish.checksums.insecure=true 13 | -------------------------------------------------------------------------------- /gradle/wrapper/.gitignore: -------------------------------------------------------------------------------- 1 | gradle-wrapper.jar 2 | GradleWrapperDownloader.class 3 | -------------------------------------------------------------------------------- /gradle/wrapper/gradle-wrapper.properties: -------------------------------------------------------------------------------- 1 | #Wed Jan 16 22:46:13 PST 2019 2 | distributionBase=GRADLE_USER_HOME 3 | distributionPath=wrapper/dists 4 | zipStoreBase=GRADLE_USER_HOME 5 | zipStorePath=wrapper/dists 6 | distributionUrl=https\://services.gradle.org/distributions/gradle-8.13-bin.zip 7 | -------------------------------------------------------------------------------- /integration/gradle: -------------------------------------------------------------------------------- 1 | ../gradle -------------------------------------------------------------------------------- /integration/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /integration/src/main/java/ai/djl/integration/tests/inference/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.inference}. */ 14 | package ai.djl.integration.tests.inference; 15 | -------------------------------------------------------------------------------- /integration/src/main/java/ai/djl/integration/tests/nn/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains tests using the engine for {@link ai.djl.nn}. */ 15 | package ai.djl.integration.tests.nn; 16 | -------------------------------------------------------------------------------- /integration/src/main/java/ai/djl/integration/tests/repository/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | /** Contains tests for {@link ai.djl.repository}. */ 14 | package ai.djl.integration.tests.repository; 15 | -------------------------------------------------------------------------------- /integration/src/main/java/ai/djl/integration/util/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains utilities for tests using the engine. */ 15 | package ai.djl.integration.util; 16 | -------------------------------------------------------------------------------- /integration/src/main/resources/log4j2.xml: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 7 | 8 | 9 | 10 | 11 | 12 | 13 | 14 | 15 | 16 | 17 | 18 | 19 | 20 | 21 | -------------------------------------------------------------------------------- /integration/src/test/java/ai/djl/integration/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains code to run the integration test suites. */ 15 | package ai.djl.integration; 16 | -------------------------------------------------------------------------------- /integration/src/test/java/ai/djl/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains integration and coverage testing code. */ 15 | package ai.djl; 16 | -------------------------------------------------------------------------------- /jacoco/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /jupyter/README.md: -------------------------------------------------------------------------------- 1 | # DJL - Jupyter notebooks 2 | 3 | The jupyter notebook documentation and examples have been moved to the [DJL Demos repo](https://docs.djl.ai/master/docs/demos/jupyter/index.html). -------------------------------------------------------------------------------- /model-zoo/build.gradle.kts: -------------------------------------------------------------------------------- 1 | import java.io.ByteArrayOutputStream 2 | 3 | plugins { 4 | ai.djl.javaProject 5 | ai.djl.publish 6 | } 7 | 8 | dependencies { 9 | api(project(":api")) 10 | 11 | testImplementation(libs.testng) 12 | testImplementation(libs.slf4j.simple) 13 | } 14 | 15 | tasks.register("syncS3") { 16 | workingDir = project.projectDir 17 | commandLine( 18 | "sh", 19 | "-c", 20 | "find . -name .DS_Store | xargs rm && aws s3 sync src/test/resources/mlrepo s3://djl-ai/mlrepo --acl public-read" 21 | ) 22 | 23 | standardOutput = ByteArrayOutputStream() 24 | ext["output"] = { standardOutput.toString() } 25 | } 26 | -------------------------------------------------------------------------------- /model-zoo/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /model-zoo/src/main/java/ai/djl/basicmodelzoo/basic/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the basic built-in models. */ 15 | package ai.djl.basicmodelzoo.basic; 16 | -------------------------------------------------------------------------------- /model-zoo/src/main/java/ai/djl/basicmodelzoo/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains the built-in {@link ai.djl.basicmodelzoo.BasicModelZoo}. */ 15 | package ai.djl.basicmodelzoo; 16 | -------------------------------------------------------------------------------- /model-zoo/src/main/javadoc/overview.html: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 |

This document is the API specification for the Deep Java Library (DJL) Model Zoo.

7 | 8 |

9 | The integration module contains a number of built-in models and translators. 10 | See here for more details. 11 |

12 | 13 | 14 | 15 | -------------------------------------------------------------------------------- /model-zoo/src/main/resources/META-INF/native-image/ai/djl/basicmodelzoo/reflect-config.json: -------------------------------------------------------------------------------- 1 | [ 2 | { 3 | "name": "ai.djl.basicmodelzoo.BasicZooProvider", 4 | "allDeclaredConstructors": true, 5 | "allPublicConstructors": true, 6 | "allDeclaredMethods": true, 7 | "allPublicMethods": true 8 | }, 9 | { 10 | "name": "ai.djl.basicmodelzoo.BasicModelZoo", 11 | "allDeclaredFields": true 12 | } 13 | ] 14 | -------------------------------------------------------------------------------- /model-zoo/src/main/resources/META-INF/native-image/ai/djl/basicmodelzoo/resource-config.json: -------------------------------------------------------------------------------- 1 | { 2 | "resources": [ 3 | { 4 | "pattern": "META-INF/services/ai.djl.repository.zoo.ZooProvider" 5 | } 6 | ] 7 | } 8 | -------------------------------------------------------------------------------- /model-zoo/src/main/resources/META-INF/services/ai.djl.repository.zoo.ZooProvider: -------------------------------------------------------------------------------- 1 | ai.djl.basicmodelzoo.BasicZooProvider 2 | -------------------------------------------------------------------------------- /model-zoo/src/test/resources/mlrepo/model/cv/image_classification/ai/djl/zoo/synset_cifar10.txt: -------------------------------------------------------------------------------- 1 | airplane 2 | automobile 3 | bird 4 | cat 5 | deer 6 | dog 7 | frog 8 | horse 9 | ship 10 | truck -------------------------------------------------------------------------------- /model-zoo/src/test/resources/mlrepo/model/cv/image_classification/ai/djl/zoo/synset_mnist.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | 6 8 | 7 9 | 8 10 | 9 -------------------------------------------------------------------------------- /model-zoo/src/test/resources/mlrepo/model/cv/object_detection/ai/djl/zoo/synset_pikachu.txt: -------------------------------------------------------------------------------- 1 | pikachu -------------------------------------------------------------------------------- /postBuild: -------------------------------------------------------------------------------- 1 | git clone https://github.com/frankfliu/IJava.git 2 | cd IJava/ 3 | ./gradlew installKernel 4 | cd .. 5 | rm -rf Ijava/ ~/.gradle 6 | -------------------------------------------------------------------------------- /serving/README.md: -------------------------------------------------------------------------------- 1 | # DJL Serving 2 | 3 | **This module has been moved to [deepjavalibrary/djl-serving](https://github.com/deepjavalibrary/djl-serving) repository.** 4 | 5 | DJL serving is a high performance universal model serving solution. You can use djl-serving serve the 6 | following models out of the box: 7 | 8 | - PyTorch TorchScript model 9 | - TensorFlow SavedModel bundle 10 | - Apache MXNet model 11 | 12 | You can install extra extensions to enable the following models: 13 | 14 | - ONNX model 15 | - XGBoost model 16 | - LightGBM model 17 | - Sentencepiece model 18 | - fastText/BlazingText model 19 | -------------------------------------------------------------------------------- /testing/README.md: -------------------------------------------------------------------------------- 1 | # DJL - Testing 2 | 3 | This folder contains Deep Java Library (DJL) testing utilities to be used by other modules. 4 | No actual tests should live within this module. 5 | 6 | -------------------------------------------------------------------------------- /testing/build.gradle.kts: -------------------------------------------------------------------------------- 1 | plugins { 2 | ai.djl.javaProject 3 | } 4 | 5 | dependencies { 6 | api(project(":api")) 7 | api(libs.testng) 8 | } 9 | 10 | tasks.compileJava { 11 | options.apply { 12 | release = 11 13 | } 14 | } 15 | -------------------------------------------------------------------------------- /testing/gradle: -------------------------------------------------------------------------------- 1 | ../gradle -------------------------------------------------------------------------------- /testing/gradlew: -------------------------------------------------------------------------------- 1 | ../gradlew -------------------------------------------------------------------------------- /testing/src/main/java/ai/djl/testing/package-info.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright 2019 Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | 14 | /** Contains testing utilities for other modules. */ 15 | package ai.djl.testing; 16 | -------------------------------------------------------------------------------- /tools/conf/licenseHeader.java: -------------------------------------------------------------------------------- 1 | /* 2 | * Copyright XXXX Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 | * 4 | * Licensed under the Apache License, Version 2.0 (the "License"). You may not use this file except in compliance 5 | * with the License. A copy of the License is located at 6 | * 7 | * http://aws.amazon.com/apache2.0/ 8 | * 9 | * or in the "license" file accompanying this file. This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES 10 | * OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions 11 | * and limitations under the License. 12 | */ 13 | -------------------------------------------------------------------------------- /tools/scripts/add_online_runner.py: -------------------------------------------------------------------------------- 1 | from pathlib import Path 2 | import glob, json 3 | 4 | constructor = {"cell_type": "markdown", "metadata": {}} 5 | prefix = "Run this notebook online:[![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/deepjavalibrary/djl/master?filepath=" 6 | 7 | for file in Path('.').glob('**/*.ipynb'): 8 | with open(file, mode= "r", encoding= "utf-8") as f: 9 | data = json.loads(f.read()) 10 | with open(file, 'w') as writer: 11 | constructor["source"] = [prefix + str(file) + ")"] 12 | data["cells"].insert(0, constructor) 13 | writer.write(json.dumps(data)) 14 | -------------------------------------------------------------------------------- /tools/scripts/broken_link_checker.sh: -------------------------------------------------------------------------------- 1 | #!/usr/bin/env bash 2 | 3 | # installation 4 | sudo apt-get update 5 | curl -sL https://deb.nodesource.com/setup_12.x | sudo -E bash - 6 | sudo apt-get install -y nodejs 7 | sudo apt-get install npm 8 | sudo npm install -g markdown-link-check 9 | sudo npm install broken-link-checker -g 10 | 11 | 12 | # check brokenlinks in md files 13 | for i in $(find ../../ -name '*.md'); 14 | do markdown-link-check "$i"; 15 | done 16 | 17 | VERSION=0.3.0 18 | # check broken link in website and java doc site 19 | blc https://djl.ai/ -ro -------------------------------------------------------------------------------- /website/img/background2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/background2.png -------------------------------------------------------------------------------- /website/img/bg/img-bg-customer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-bg-customer.jpg -------------------------------------------------------------------------------- /website/img/bg/img-bg-feature.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-bg-feature.jpeg -------------------------------------------------------------------------------- /website/img/bg/img-bg-sky.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-bg-sky.jpg -------------------------------------------------------------------------------- /website/img/bg/img-bg-twitter.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-bg-twitter.jpg -------------------------------------------------------------------------------- /website/img/bg/img-bg5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-bg5.jpg -------------------------------------------------------------------------------- /website/img/bg/img-customer-bg.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/bg/img-customer-bg.jpg -------------------------------------------------------------------------------- /website/img/community_leader/anthony.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/anthony.jpeg -------------------------------------------------------------------------------- /website/img/community_leader/aziz.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/aziz.jpg -------------------------------------------------------------------------------- /website/img/community_leader/christoph.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/christoph.png -------------------------------------------------------------------------------- /website/img/community_leader/erik.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/erik.png -------------------------------------------------------------------------------- /website/img/community_leader/frank.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/frank.png -------------------------------------------------------------------------------- /website/img/community_leader/fyz.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/fyz.jpg -------------------------------------------------------------------------------- /website/img/community_leader/jake.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/jake.jpg -------------------------------------------------------------------------------- /website/img/community_leader/keerthan.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/keerthan.jpg -------------------------------------------------------------------------------- /website/img/community_leader/kimi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/kimi.png -------------------------------------------------------------------------------- /website/img/community_leader/lai.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/lai.jpg -------------------------------------------------------------------------------- /website/img/community_leader/lu.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/lu.png -------------------------------------------------------------------------------- /website/img/community_leader/marat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/marat.png -------------------------------------------------------------------------------- /website/img/community_leader/marcos.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/marcos.png -------------------------------------------------------------------------------- /website/img/community_leader/qing.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/qing.jpeg -------------------------------------------------------------------------------- /website/img/community_leader/stanislav.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/stanislav.png -------------------------------------------------------------------------------- /website/img/community_leader/wei.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/wei.jpg -------------------------------------------------------------------------------- /website/img/community_leader/zach.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/community_leader/zach.jpeg -------------------------------------------------------------------------------- /website/img/deepjavalibrary.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/deepjavalibrary.png -------------------------------------------------------------------------------- /website/img/djl-large.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/djl-large.png -------------------------------------------------------------------------------- /website/img/djl-middle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/djl-middle.png -------------------------------------------------------------------------------- /website/img/djl.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/djl.png -------------------------------------------------------------------------------- /website/img/favicon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/favicon.png -------------------------------------------------------------------------------- /website/img/stars-in-the-night-sky.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deepjavalibrary/djl/1fc3cd0c5dc692bbd333401cc3664144f5c91deb/website/img/stars-in-the-night-sky.jpg -------------------------------------------------------------------------------- /website/js/init.js: -------------------------------------------------------------------------------- 1 | (function($){ 2 | $(function(){ 3 | 4 | $('.sidenav').sidenav(); 5 | $('.parallax').parallax(); 6 | $('.tabs').tabs(); 7 | $('.collapsible').collapsible(); 8 | $('.slider').slider(); 9 | }); // end of document ready 10 | })(jQuery); // end of jQuery name space 11 | --------------------------------------------------------------------------------