dfdx

Spark.jl

Julia binding for Apache Spark

Yota.jl

Reverse-mode automatic differentiation in Julia

Avalon.jl

Starter kit for legendary models

Espresso.jl

Expression transformation package

Ghost.jl

The Code Tracer

Lilith.jl

Renamed to Avalon.jl

HQDL.jl

Curated list of high-quality operators for deep learning in Julia

RDKafka.jl

Wrapper for librdkafka

Umlaut.jl

The Code Tracer

NaiveBayes.jl

Naive Bayes classifier

Leto.jl

Remix.jl

Deep learning primitives, restricted and robust

dfdx.github.io

df/dx

NNlib.jl

ChainRules.jl

forward and reverse mode automatic differentiation primitives for Julia Base + StdLibs

ChainRulesTestUtils.jl

Utilities for testing custom AD primitives.

CompilerPluginTools.jl

collective tools for Julia SSA IR.

CUDA.jl

CUDA programming in Julia.

fabrique

Research-friendly implementations of LLMs in JAX

Fusion.jl

gendo

Generative models

IRTools.jl

Mike's Little Intermediate Representation

kava

Generative search engine

llama-2-jax

JAX implementation of the Llama 2 model

Mixtape.jl

A static method overlay and optimization tool with configurable code generation and execution.

MLServe.jl

Web server for ML models

onnx

Open standard for machine learning interoperability

ONNX.jl

Read ONNX graphs in Julia

ScatterNNlib.jl

A scatter operation library for neural network

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.