best-of-ml-julia

🏆 A ranked list of awesome machine learning Julia libraries. Updated weekly.

This curated list contains 100 awesome open-source projects with a total of 46K stars grouped into 8 categories. All projects are ranked by a project-quality score, which is calculated based on various metrics automatically collected from GitHub and different package managers. If you like to add or update projects, feel free to open an issue, submit a pull request, or directly edit the projects.yaml. Contributions are very welcome!

🧙‍♂️ Discover other best-of lists or create your own.

Machine Learning Frameworks 63 projects
Text Data & NLP 12 projects
Image Data 3 projects
Data Loading & Extraction 7 projects
Workflow & Experiment Tracking 1 projects
Hyperparameter Optimization & AutoML 1 projects
Probabilistics & Statistics 12 projects
Model Interpretability 1 projects
Others 2 projects

Explanation

🥇🥈🥉 Combined project-quality score
⭐️ Star count from GitHub
🐣 New project (less than 6 months old)
💤 Inactive project (6 months no activity)
💀 Dead project (12 months no activity)
📈📉 Project is trending up or down
➕ Project was recently added
❗️ Warning (e.g. missing/risky license)
👨‍💻 Contributors count from GitHub
🔀 Fork count from GitHub
📋 Issue count from GitHub
⏱️ Last update timestamp on package manager
📥 Download count from package manager
📦 Number of dependent projects

Machine Learning Frameworks

General-purpose machine learning and deep learning frameworks.

incubator-mxnet (🥇31 · ⭐ 19K) - Lightweight, Portable, Flexible Distributed/Mobile Deep.. Apache-2

GitHub (👨‍💻 960 · 🔀 6.5K · 📥 24K · 📦 2K · 📋 9.3K - 18% open · ⏱️ 21.05.2021):
```
 git clone https://github.com/apache/incubator-mxnet
```

Turing.jl (🥇20 · ⭐ 1.2K) - Bayesian inference with probabilistic programming. MIT

GitHub (👨‍💻 74 · 🔀 150 · 📋 820 - 19% open · ⏱️ 20.05.2021):
```
 git clone https://github.com/TuringLang/Turing.jl
```

Gen.jl (🥇19 · ⭐ 1.5K) - A general-purpose probabilistic programming system with programmable.. Apache-2

GitHub (👨‍💻 29 · 🔀 140 · 📋 240 - 54% open · ⏱️ 13.05.2021):
```
 git clone https://github.com/probcomp/Gen.jl
```

DiffEqFlux.jl (🥇18 · ⭐ 510) - Universal neural differential equations with O(1) backprop, GPUs,.. MIT

GitHub (👨‍💻 52 · 🔀 99 · 📋 240 - 23% open · ⏱️ 20.05.2021):
```
 git clone https://github.com/SciML/DiffEqFlux.jl
```

DFTK.jl/ (🥈16 · ⭐ 160) - Density-functional toolkit. MIT

GitHub (🔀 25 · 📋 120 - 39% open · ⏱️ 12.05.2021):
```
 git clone https://github.com/JuliaMolSim/DFTK.jl/
```

MixedModels.jl (🥈15 · ⭐ 260) - A Julia package for fitting (statistical) mixed-effects models. MIT

GitHub (👨‍💻 23 · 🔀 37 · 📋 220 - 9% open · ⏱️ 16.05.2021):
```
 git clone https://github.com/JuliaStats/MixedModels.jl
```

AutoMLPipeline.jl (🥈15 · ⭐ 250) - A package that makes it trivial to create and evaluate.. MIT

GitHub (👨‍💻 4 · 🔀 18 · 📋 64 - 28% open · ⏱️ 23.05.2021):
```
 git clone https://github.com/IBM/AutoMLPipeline.jl
```

GeometricFlux.jl (🥈15 · ⭐ 200) - Geometric Deep Learning for Flux. MIT

GitHub (👨‍💻 10 · 🔀 17 · 📋 49 - 44% open · ⏱️ 17.05.2021):
```
 git clone https://github.com/yuehhua/GeometricFlux.jl
```

ADCME.jl (🥈15 · ⭐ 130) - Automatic Differentiation Library for Computational and Mathematical.. MIT

GitHub (👨‍💻 4 · 🔀 35 · 📥 96 · 📋 37 - 37% open · ⏱️ 21.05.2021):
```
 git clone https://github.com/kailaix/ADCME.jl
```

KernelFunctions.jl (🥈14 · ⭐ 120) - Julia package for kernel functions for machine learning. MIT

GitHub (👨‍💻 14 · 🔀 14 · 📋 100 - 37% open · ⏱️ 15.05.2021):
```
 git clone https://github.com/JuliaGaussianProcesses/KernelFunctions.jl
```

FastAI.jl (🥉12 · ⭐ 160) - Repository of best practices for deep learning in Julia, inspired by.. MIT

GitHub (👨‍💻 11 · 🔀 20 · 📋 18 - 50% open · ⏱️ 23.05.2021):
```
 git clone https://github.com/FluxML/FastAI.jl
```

Show 52 hidden projects...

Flux.jl (🥇19 · ⭐ 3K) - Relax! Flux is the ML library that doesn't make you tensor. ❗Unlicensed
Knet.jl (🥇19 · ⭐ 1.3K) - Ko University deep learning framework. ❗Unlicensed
MLJBase.jl (🥇17 · ⭐ 63) - Core functionality for the MLJ machine learning framework. MIT
DifferentialEquations.jl (🥈16 · ⭐ 1.8K) - Multi-language suite for high-performance solvers of.. ❗Unlicensed
MLJ.jl (🥈16 · ⭐ 1.1K) - A Julia machine learning framework. ❗Unlicensed
Zygote.jl (🥈16 · ⭐ 980) - Intimate Affection Auditor. ❗Unlicensed
ModelingToolkit.jl (🥈16 · ⭐ 630) - A modeling framework for automatically parallelized.. ❗Unlicensed
OrdinaryDiffEq.jl (🥈16 · ⭐ 220) - High performance differential equation solvers for.. ❗Unlicensed
TensorFlow.jl (🥈15 · ⭐ 830) - A Julia wrapper for TensorFlow. ❗Unlicensed
NeuralPDE.jl (🥈15 · ⭐ 330) - Physics-Informed Neural Networks (PINN) and Deep BSDE.. ❗Unlicensed
Distances.jl (🥈15 · ⭐ 270) - A Julia package for evaluating distances (metrics) between.. ❗Unlicensed
ReinforcementLearning.jl (🥈15 · ⭐ 220) - A reinforcement learning package for Julia. ❗Unlicensed
DiffEqBase.jl (🥈15 · ⭐ 120) - The lightweight Base library for shared types and.. ❗Unlicensed
ScikitLearn.jl (🥈14 · ⭐ 440) - Julia implementation of the scikit-learn API.. ❗Unlicensed
Clustering.jl (🥈14 · ⭐ 230) - A Julia package for data clustering. ❗Unlicensed
DecisionTree.jl (🥈14 · ⭐ 230 · 💤) - Julia implementation of Decision Tree (CART) and.. ❗Unlicensed
XGBoost.jl (🥈14 · ⭐ 180) - XGBoost Julia Package. ❗Unlicensed
MLBase.jl (🥈14 · ⭐ 180 · 💀) - A set of functions to support the development of machine learning.. MIT
NNlib.jl (🥈14 · ⭐ 99) - Neural Network primitives with multiple backends. ❗Unlicensed
ForneyLab.jl (🥈14 · ⭐ 90) - Julia package for automatically generating Bayesian.. ❗Unlicensed
Yota.jl (🥈14 · ⭐ 84) - Reverse-mode automatic differentiation in Julia. MIT
ReservoirComputing.jl (🥈14 · ⭐ 66) - Reservoir computing utilities for scientific machine.. MIT
TSML.jl (🥈14 · ⭐ 59) - A package for time series data processing, classification, clustering, and.. MIT
FluxTraining.jl (🥈14 · ⭐ 34) - A flexible neural net training library inspired by fast.ai. MIT
Mocha.jl (🥉13 · ⭐ 1.3K · 💀) - Deep Learning framework for Julia. ❗Unlicensed
Yao.jl (🥉13 · ⭐ 540 · 📈) - Extensible, Efficient Quantum Algorithm Design for Humans. ❗Unlicensed
Stheno.jl (🥉13 · ⭐ 250) - Probabilistic Programming with Gaussian processes in Julia. ❗Unlicensed
ArrayFire.jl (🥉13 · ⭐ 180) - Julia wrapper for the ArrayFire library. ❗Unlicensed
Merlin.jl (🥉13 · ⭐ 150 · 💀) - Deep Learning for Julia. MIT
Enzyme.jl (🥉13 · ⭐ 96) - Julia bindings for the Enzyme automatic differentiator. MIT
LossFunctions.jl (🥉13 · ⭐ 90 · 📈) - Julia package of loss functions for machine learning. ❗Unlicensed
UMAP.jl (🥉13 · ⭐ 81) - Uniform Manifold Approximation and Projection (UMAP) implementation in Julia. MIT
Mads.jl (🥉13 · ⭐ 75) - MADS: Model Analysis & Decision Support. ❗Unlicensed
Loess.jl (🥉13 · ⭐ 54) - Local regression, so smooooth!. ❗Unlicensed
Mamba.jl (🥉12 · ⭐ 240 · 💤) - Markov chain Monte Carlo (MCMC) for Bayesian analysis in.. ❗Unlicensed
Reinforce.jl (🥉12 · ⭐ 180) - Abstractions, algorithms, and utilities for reinforcement.. ❗Unlicensed
AutoGrad.jl (🥉12 · ⭐ 150) - Julia port of the Python autograd package. ❗Unlicensed
DiffEqBayes.jl (🥉12 · ⭐ 93) - Extension functionality which uses Stan.jl,.. ❗Unlicensed
NMF.jl (🥉12 · ⭐ 68) - A Julia package for non-negative matrix factorization. ❗Unlicensed
CausalInference.jl (🥉12 · ⭐ 64) - Causal inference, graphical models and structure.. ❗Unlicensed
JLBoost.jl (🥉12 · ⭐ 61) - A 100%-Julia implementation of Gradient-Boosting Regression Tree.. MIT
bat (🥉12 · ⭐ 56) - Bayesian analysis toolkit http://mpp.mpg.de/bat. ❗Unlicensed
DiffEqParamEstim.jl (🥉12 · ⭐ 33) - Easy scientific machine learning (SciML) parameter.. ❗Unlicensed
AtomicGraphNets.jl (🥉12 · ⭐ 25) - Atomic graph models for molecules and crystals in Julia. MIT
Torch.jl (🥉11 · ⭐ 130) - Sensible extensions for exposing torch in Julia. ❗Unlicensed
ONNX.jl (🥉11 · ⭐ 80) - Read ONNX graphs in Julia. ❗Unlicensed
ROCAnalysis.jl (🥉11 · ⭐ 24) - Receiver Operating Characteristics and functions for.. ❗Unlicensed
ManifoldLearning.jl (🥉10 · ⭐ 59) - A Julia package for manifold learning and nonlinear.. ❗Unlicensed
lathe (🥉10 · ⭐ 44) - A tool for generating bacterial genomes from metagenomes with nanopore long.. MIT
DiscriminantAnalysis.jl (🥉9 · ⭐ 8) - Regularized discriminant analysis in Julia. MIT
TimeSeriesClassification.jl (🥉8 · ⭐ 19) - Machine Learning with Time Series in Julia. MIT
OpenAIGym.jl (🥉7 · ⭐ 84 · 💤) - OpenAI's Gym binding for Julia. ❗Unlicensed

Text Data & NLP

Libraries for processing, cleaning, manipulating, and analyzing text data as well as for diverse NLP tasks.

Transformers.jl (🥇16 · ⭐ 190) - Julia Implementation of Transformer models. MIT

GitHub (👨‍💻 8 · 🔀 29 · 📋 21 - 14% open · ⏱️ 22.05.2021):
```
 git clone https://github.com/chengchingwen/Transformers.jl
```

Show 11 hidden projects...

TextAnalysis.jl (🥇17 · ⭐ 300) - Julia package for text analysis. ❗Unlicensed
WordTokenizers.jl (🥈12 · ⭐ 68) - High performance tokenizers for natural language.. ❗Unlicensed
AdaGram.jl (🥈11 · ⭐ 160 · 💀) - Adaptive Skip-gram implementation in Julia. MIT
Embeddings.jl (🥈11 · ⭐ 52 · 💀) - Functions and data dependencies for loading various word.. MIT
BytePairEncoding.jl (🥈11 · ⭐ 3) - Julia implementation of Byte Pair Encoding for NLP. MIT
Languages.jl (🥉10 · ⭐ 34) - A package for working with human languages. ❗Unlicensed
TextModels.jl (🥉10 · ⭐ 7) - Neural Network based models for Natural Language Processing. ❗Unlicensed
Word2Vec.jl (🥉9 · ⭐ 51) - Julia interface to word2vec. ❗Unlicensed
WordNet.jl (🥉9 · ⭐ 22 · 💀) - A Julia package for Princeton's WordNet. ❗Unlicensed
TopicModels.jl (🥉8 · ⭐ 31 · 💤) - TopicModels for Julia. MIT
Snowball.jl (🥉5 · ⭐ 1) - Snowball stemming algorithms. MIT

Image Data

Libraries for image & video processing, manipulation, and augmentation as well as libraries for computer vision.

Show 3 hidden projects...

Images.jl (🥇16 · ⭐ 440) - An image library for Julia. ❗Unlicensed
Metalhead.jl (🥉15 · ⭐ 200 · 📈) - Computer vision models for Flux. ❗Unlicensed
Augmentor.jl (🥉13 · ⭐ 99 · 📈) - A fast image augmentation library in Julia for machine.. ❗Unlicensed

Data Loading & Extraction

Libraries for loading, collecting, and extracting data from a variety of data sources and formats.

MLDatasets.jl (🥈15 · ⭐ 120) - Utility package for accessing common Machine Learning datasets in.. MIT

GitHub (👨‍💻 13 · 🔀 25 · 📥 4 · 📋 25 - 44% open · ⏱️ 30.03.2021):
```
 git clone https://github.com/JuliaML/MLDatasets.jl
```

Show 6 hidden projects...

DataFrames.jl (🥇18 · ⭐ 1K) - In-memory tabular data in Julia. ❗Unlicensed
DataDeps.jl (🥈13 · ⭐ 91) - reproducible data setup for reproducible science. ❗Unlicensed
DataAugmentation.jl (🥈13 · ⭐ 4) - Flexible data augmentation library for machine and deep.. MIT
JuliaML/MLDataUtils.jl (🥉12 · ⭐ 77) - Utility package for generating, loading, splitting,.. ❗Unlicensed
CorpusLoaders.jl (🥉11 · ⭐ 21) - A variety of loaders for various NLP corpora. ❗Unlicensed
MLDataPattern.jl (🥉9 · ⭐ 40) - Utility package for subsetting, resampling, iteration,.. ❗Unlicensed

Workflow & Experiment Tracking

Libraries to organize, track, and visualize machine learning experiments.

Show 1 hidden projects...

TensorBoardLogger.jl (🥇14 · ⭐ 69) - Easy peasy logging to TensorBoard with Julia. MIT

Hyperparameter Optimization & AutoML

Libraries for hyperparameter optimization, automl and neural architecture search.

Show 1 hidden projects...

MLJTuning.jl (🥇14 · ⭐ 35) - Hyperparameter optimization algorithms for use in the MLJ machine.. MIT

Probabilistics & Statistics

Libraries providing capabilities for probabilistic programming/reasoning, bayesian inference, gaussian processes, or statistics.

Show 12 hidden projects...

Distributions.jl (🥇18 · ⭐ 690) - A Julia package for probability distributions and.. ❗Unlicensed
GLM.jl (🥇15 · ⭐ 380) - Generalized linear models in Julia. ❗Unlicensed
HypothesisTests.jl (🥇15 · ⭐ 180) - Hypothesis tests for Julia. ❗Unlicensed
MultivariateStats.jl (🥈14 · ⭐ 230) - A Julia package for multivariate statistics and data.. ❗Unlicensed
GaussianProcesses.jl (🥈14 · ⭐ 220) - A Julia package for Gaussian Processes. ❗Unlicensed
MeasureTheory.jl (🥈14 · ⭐ 170) - MIT
ReadStat.jl (🥈14 · ⭐ 66) - Read files from Stata, SAS, and SPSS. MIT
KernelDensity.jl (🥉13 · ⭐ 100) - Kernel density estimators for Julia. ❗Unlicensed
Omega.jl (🥉13 · ⭐ 86) - Causal, Higher-Order, Probabilistic Programming. MIT
GaussianMixtures.jl (🥉13 · ⭐ 59) - Large scale Gaussian Mixture Models. ❗Unlicensed
GLMNet.jl (🥉12 · ⭐ 79) - Julia wrapper for fitting Lasso/ElasticNet GLM models using.. ❗Unlicensed
cscherrer/SossMLJ.jl (🥉11 · ⭐ 12) - SossMLJ makes it easy to build MLJ machines from user-.. MIT

Model Interpretability

Libraries to visualize, explain, debug, evaluate, and interpret machine learning models.

Show 1 hidden projects...

Fairness.jl (🥇12 · ⭐ 18) - Julia Toolkit with fairness metrics and bias mitigation algorithms. MIT

Others

Show 2 hidden projects...

ReinforcementLearningZoo.jl (🥇13 · ⭐ 45) - MIT
DistributedReinforcementLearning.jl (🥉6 · ⭐ 2) - If it works, it works everywhere!. MIT

Related Resources

best-of-ml-python: A ranked list of awesome machine learning Python libraries.
best-of-ml-rust: A ranked list of awesome machine learning Rust libraries.

Contribution

Contributions are encouraged and always welcome! If you like to add or update projects, choose one of the following ways:

Open an issue by selecting one of the provided categories from the issue page and fill in the requested information.
Modify the projects.yaml with your additions or changes, and submit a pull request. This can also be done directly via the Github UI.

If you like to contribute to or share suggestions regarding the project metadata collection or markdown generation, please refer to the best-of-generator repository. If you like to create your own best-of list, we recommend to follow this guide.

For more information on how to add or update projects, please read the contribution guidelines. By participating in this project, you agree to abide by its Code of Conduct.

Toni-zgz/best-of-ml-julia

best-of-ml-julia

Contents

Explanation

Machine Learning Frameworks

Text Data & NLP

Image Data

Data Loading & Extraction

Workflow & Experiment Tracking

Hyperparameter Optimization & AutoML

Probabilistics & Statistics

Model Interpretability

Others

Related Resources

Contribution

License