Jax Js Alternatives

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

Foolbox is a Python library for running adversarial attacks against machine learning models. It takes a trained model, wraps it in a framework specific adapter, and applies an attack to a batch of inputs across a range of perturbation budgets, returning the adversarial examples and a success flag for each one. The purp...

A retargetable MLIR-based machine learning compiler and runtime toolkit.

IREE, short for Intermediate Representation Execution Environment, is an MLIR based compiler and runtime that lowers machine learning models to a unified intermediate representation. The same toolchain is meant to scale from datacenter workloads down to the constraints of mobile and edge deployments, with binary sizes ...

The nnsight package enables interpreting and manipulating the internals of deep learned models.

NNsight is a Python library for interpreting and manipulating the internals of deep learning models. It gives researchers a Pythonic interface to read activations at any layer during a forward pass, modify those activations to study causal effects, and compute gradients with respect to intermediate values. It was devel...

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

ONNX Runtime is an open source engine for running and accelerating machine learning models across platforms. It executes models exported to the ONNX format and applies built in optimizations to improve latency, throughput, memory use and binary size on CPUs, GPUs and NPUs. It is aimed at developers who need to ship tr...

A front-end GUI for interacting with the AI Horde / Stable Diffusion distributed cluster

ArtBot is a free web interface for generating images with Stable Diffusion through the AI Horde, a distributed open source network of volunteer GPUs. Instead of running models locally or paying for a hosted service, users submit prompts in the browser and the request is picked up by a worker on the Horde, with results ...

An autoML framework & toolkit for machine learning on graphs.

AutoGL is an open source automated machine learning toolkit for graph data, developed by the Tsinghua University MN Lab (THUMNLab). It lets researchers and developers run AutoML on graph datasets and tasks: rather than hand picking a graph neural network and tuning it, you hand AutoGL a dataset and a task and its solve...