AI Packages

Flux.jl4122Relax! Flux is the ML library that doesn't make you tensor

MLJ.jl1589A Julia machine learning framework

Knet.jl1403Koç University deep learning framework.

Mocha.jl1283Deep Learning framework for Julia

AlphaZero.jl1131A generic, simple and fast implementation of Deepmind's AlphaZero algorithm.

BrainFlow.jl935BrainFlow is a library intended to obtain, parse and analyze EEG, EMG, ECG and other kinds of data from biosensors

TensorFlow.jl866A Julia wrapper for TensorFlow

DiffEqFlux.jl771Universal neural differential equations with O(1) backprop, GPUs, and stiff+nonstiff DE solvers, demonstrating scientific machine learning (SciML) and physicsinformed machine learning methods

NeuralNetDiffEq.jl755PhysicsInformed Neural Networks (PINN) and Deep BSDE Solvers of Differential Equations for Scientific Machine Learning (SciML) accelerated simulation

FastAI.jl557Repository of best practices for deep learning in Julia, inspired by fastai

ScikitLearn.jl520Julia implementation of the scikitlearn API https://cstjean.github.io/ScikitLearn.jl/dev/

ReinforcementLearning.jl498A reinforcement learning package for Julia

Transformers.jl420Julia Implementation of Transformer models

MXNet.jl371MXNet Julia Package  flexible and efficient deep learning in Julia

StatisticalRethinking.jl366Julia package with selected functions in the R package `rethinking`. Used in the SR2... projects.

TextAnalysis.jl352Julia package for text analysis

AutoMLPipeline.jl325A package that makes it trivial to create and evaluate machine learning pipeline architectures.

DecisionTree.jl316Julia implementation of Decision Tree (CART) and Random Forest algorithms

Enzyme.jl311Julia bindings for the Enzyme automatic differentiator

Clustering.jl311A Julia package for data clustering

Lux.jl300Explicitly Parameterized Neural Networks in Julia

Metalhead.jl297Computer vision models for Flux

Dojo.jl221A differentiable physics engine for robotics

MLDatasets.jl204Utility package for accessing common Machine Learning datasets in Julia

Reinforce.jl202Abstractions, algorithms, and utilities for reinforcement learning in Julia

SimpleChains.jl195Simple chains

AbstractGPs.jl192Abstract types and methods for Gaussian Processes.

Torch.jl183Sensible extensions for exposing torch in Julia.

MLBase.jl179A set of functions to support the development of machine learning algorithms

ReservoirComputing.jl172Reservoir computing utilities for scientific machine learning (SciML)

AdaGram.jl169Adaptive Skipgram implementation in Julia

NNlib.jl165Neural Network primitives with multiple backends

GraphNeuralNetworks.jl153Graph Neural Networks in Julia

Yota.jl145Reversemode automatic differentiation in Julia

Merlin.jl144Deep Learning for Julia

EvoTrees.jl143Boosted trees in Julia

SeaPearl.jl141Julia hybrid constraint programming solver enhanced by a reinforcement learning driven search.

MLJBase.jl140Core functionality for the MLJ machine learning framework

LossFunctions.jl137Julia package of loss functions for machine learning.

ForneyLab.jl135Julia package for automatically generating Bayesian inference algorithms through message passing on Forneystyle factor graphs.
Loading more...