A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extraction, and more.
-
Updated
Jun 2, 2024 - Python
A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extraction, and more.
Deep Learning for humans
Factor graphs and nonlinear least squares for JAX
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
Additive manufacturing simulation with JAX.
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
PEP 503 repository index for jax[cuda]
NGC-Learn: Neurobiological Learning and Biomimetic Systems Simulation in Python
Machine learning algorithms for many-body quantum systems
Accelerate your training with this open-source library. Optimize performance with streamlined training and serving options with JAX. 🚀
PennyLane is a cross-platform Python library for quantum computing, quantum machine learning, and quantum chemistry. Train a quantum computer the same way as a neural network.
Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)
🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.
Coordinates in JAX
This is a JAX/Flax-based transformer language model trained on a Japanese dataset. It is based on the official Flax example code (lm1b).
Data processing utilities in keras3
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
Add a description, image, and links to the jax topic page so that developers can more easily learn about it.
To associate your repository with the jax topic, visit your repo's landing page and select "manage topics."