jax

JAX-based neural network library

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023

Training and serving large-scale neural networks with auto parallelization.

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

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

🔮 A refreshing functional take on deep learning, compatible with your favorite libraries

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

A simple, performant, and scalable Jax LLM!

An Extensible Deep Learning Library

Distributed AI Model Training and LLM Fine-Tuning on Kubernetes

JAX - A curated list of resources https://github.com/google/jax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

A deep learning package for many-body potential energy representation and molecular dynamics

A JAX research toolkit for building, editing, and visualizing neural networks.

Automated Machine Learning on Kubernetes

Long Range Arena for Benchmarking Efficient Transformers

Machine learning Guide. Learn all about Machine Learning Tools, Libraries, Frameworks, Large Language Models (LLMs), and Training Models.

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

Fine-tune and evaluate Whisper models for Automatic Speech Recognition (ASR) on custom datasets or datasets from huggingface.

Train very large language models in Jax.