An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

proxy.golang.org "jax" keyword

View the packages on the proxy.golang.org package registry that are tagged with the "jax" keyword.

Top 5.6% on proxy.golang.org
github.com/dbraun/dawdreamer v0.8.3
Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX,...
28 versions - Latest release: over 1 year ago - 1,070 stars on GitHub
Top 3.7% on proxy.golang.org
github.com/google/iree v3.9.0+incompatible
A retargetable MLIR-based machine learning compiler and runtime toolkit.
9 versions - Latest release: about 1 month ago - 2,924 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/i-m-iron-man/foragax v0.0.5
An Agent-Based Modelling framework in JAX
1 version - Latest release: over 1 year ago - 4 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/deepmind/dm_pix v0.4.4
PIX is an image processing library in JAX, for JAX.
13 versions - Latest release: 11 months ago - 353 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/mpi4jax/mpi4jax v0.8.1
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
49 versions - Latest release: 5 months ago - 496 stars on GitHub
Top 6.1% on proxy.golang.org
github.com/wandb/wandb/experimental/go-sdk
Package client implements the W&B Go SDK.
Latest release: 1 day ago - 10,440 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/PennyLaneAI/catalyst v0.13.0
A JIT compiler for hybrid quantum programs in PennyLane
23 versions - Latest release: 3 months ago - 188 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/patrick-kidger/jaxtyping v0.3.5 💰
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...
47 versions - Latest release: 2 days ago - 1,408 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/iree-org/iree v3.9.0+incompatible
A retargetable MLIR-based machine learning compiler and runtime toolkit.
9 versions - Latest release: about 1 month ago - 3,426 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/dunnolab/xland-minigrid v0.9.2
JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️
7 versions - Latest release: 22 days ago - 298 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/ikostrikov/jax-rl v0.0.6
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action sp...
1 version - Latest release: over 4 years ago - 584 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/adaptive-intelligent-robotics/QDax v0.5.0
Accelerated Quality-Diversity
11 versions - Latest release: 7 months ago - 318 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/google/tree-math v0.2.1
Mathematical operations for JAX pytrees
3 versions - Latest release: almost 2 years ago - 200 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/instadeepai/jumanji v1.1.1
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
18 versions - Latest release: 7 months ago - 769 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/thomaspinder/gpjax v0.13.4
Gaussian processes in JAX.
50 versions - Latest release: about 1 month ago - 387 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/thomaspinder/GPJax v0.13.4
Gaussian processes in JAX.
50 versions - Latest release: about 1 month ago - 387 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/danielward27/flowjax v17.2.1+incompatible
60 versions - Latest release: 3 months ago - 205 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/pyro-ppl/numpyro v0.2.4
Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.
6 versions - Latest release: almost 6 years ago - 2,533 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/vicariousinc/PGMax v0.4.1
Loopy belief propagation for factor graphs on discrete variables, in JAX!
7 versions - Latest release: over 3 years ago - 63 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/google/brax v0.14.0
Massively parallel rigidbody physics simulation on accelerator hardware.
36 versions - Latest release: 22 days ago - 2,886 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/pennylaneai/catalyst v0.13.0
A JIT compiler for hybrid quantum programs in PennyLane
23 versions - Latest release: 3 months ago - 188 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/adaptive-intelligent-robotics/qdax v0.5.0
Accelerated Quality-Diversity
11 versions - Latest release: 7 months ago - 319 stars on GitHub
Top 7.6% on proxy.golang.org
github.com/wandb/wandb/experimental/client-go
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models ...
Latest release: 5 days ago - 10,440 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/synsense/rockpool v3.0.3+incompatible
A machine learning library for spiking neural networks. Supports training with both torch and jax...
20 versions - Latest release: 21 days ago - 74 stars on GitHub
Top 4.0% on proxy.golang.org
github.com/probml/pyprobml v0.1.0
Python code for "Probabilistic Machine learning" book by Kevin Murphy
1 version - Latest release: over 3 years ago - 1 dependent repositories - 6,897 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/pasqal-io/horqrux v0.9.2
Jax-based quantum state vector simulator.
13 versions - Latest release: 7 months ago - 20 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/gchq/coreax v1.0.0
A library for coreset algorithms, written in JAX for fast execution and GPU support.
7 versions - Latest release: 10 months ago - 34 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/i-m-iron-man/Foragax v0.0.5
An Agent-Based Modelling framework in JAX
1 version - Latest release: over 1 year ago - 5 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/google/orbax v0.11.30
Orbax provides common checkpointing and persistence utilities for JAX users
69 versions - Latest release: about 1 month ago - 439 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/EdanToledo/Stoix v0.0.3
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in...
3 versions - Latest release: 4 months ago - 364 stars on GitHub
Top 9.6% on proxy.golang.org
github.com/fchollet/keras v3.13.0+incompatible
Deep Learning for humans
60 versions - Latest release: 20 days ago - 60,200 stars on GitHub
Top 9.6% on proxy.golang.org
github.com/huggingface/pytorch-transformers v4.57.3+incompatible
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
232 versions - Latest release: about 1 month ago - 119,150 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/huggingface/pytorch-pretrained-bert v4.57.3+incompatible
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
232 versions - Latest release: about 1 month ago - 126,450 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/huggingface/pytorch-pretrained-BERT v4.57.3+incompatible
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
232 versions - Latest release: about 1 month ago - 126,450 stars on GitHub
Top 3.7% on proxy.golang.org
github.com/kubeflow/trainer v1.9.3
Distributed AI Model Training and Fine-Tuning on Kubernetes
45 versions - Latest release: 6 months ago - 1,951 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/kubeflow/trainer/v2 v2.1.0
Distributed AI Model Training and Fine-Tuning on Kubernetes
7 versions - Latest release: 2 months ago - 1,951 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/deepmind/jraph v0.0.2-dev
A Graph Neural Network Library in Jax
1 version - Latest release: about 4 years ago - 1,424 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/JabinGP/mdout v0.0.0-20230330134059-d047b005eae0
一个Go语言实现的Markdown转PDF命令行工具,基于headless chrome,简单、可靠、易安装、可定制化、易拓展
2 versions - Latest release: almost 3 years ago - 102 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/sciml/scimlbenchmarks.jl v0.1.3 💰
Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solve...
4 versions - Latest release: over 2 years ago - 330 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/edantoledo/stoix v0.0.3
🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in...
3 versions - Latest release: 4 months ago - 364 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/bahremsd/tmmax v1.1.4
A fast transfer matrix method written in jax for modelling optical multilayer thin films
3 versions - Latest release: 3 months ago - 26 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/deepmind/haiku v0.0.16
JAX-based neural network library
18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/ucl-bug/jaxdf v0.2.4
A JAX-based research framework for writing differentiable numerical simulators with arbitrary dis...
6 versions - Latest release: about 3 years ago - 130 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/patrick-kidger/equinox v0.13.2 💰
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
53 versions - Latest release: 3 months ago - 2,522 stars on GitHub
Top 4.0% on proxy.golang.org
github.com/google/trax v1.4.1
Trax — Deep Learning with Clear Code and Speed
18 versions - Latest release: about 4 years ago - 1 dependent repositories - 8,287 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/ikostrikov/jaxrl v0.0.6
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action sp...
1 version - Latest release: over 4 years ago - 597 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/deepmind/PGMax v0.6.1
Loopy belief propagation for factor graphs on discrete variables in JAX
4 versions - Latest release: over 2 years ago - 150 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/SciML/SciMLBenchmarks.jl v0.1.3 💰
Scientific machine learning (SciML) benchmarks, AI for science, and (differential) equation solve...
4 versions - Latest release: over 2 years ago - 332 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/SymJAX/SymJAX v0.5.0
Documentation:
6 versions - Latest release: about 5 years ago - 120 stars on GitHub
Top 4.1% on proxy.golang.org
github.com/unifyai/ivy v1.1.17
Unified AI
17 versions - Latest release: over 2 years ago - 11,842 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/general-molecular-simulations/so3lr v0.1.0
SO3krates and Universal Pairwise Force Field for Molecular Simulation
2 versions - Latest release: about 2 months ago - 132 stars on GitHub
Top 3.5% on proxy.golang.org
github.com/kubeflow/katib v0.19.0
Automated Machine Learning on Kubernetes
29 versions - Latest release: 2 months ago - 2 dependent repositories - 1,626 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/netket/netket v3.21.0+incompatible
Machine learning algorithms for many-body quantum systems
55 versions - Latest release: 24 days ago - 625 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/deepmind/dm-haiku v0.0.16
JAX-based neural network library
18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/lanl/scico v0.0.7
Scientific Computational Imaging COde
7 versions - Latest release: 29 days ago - 141 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/brentyi/jaxlie v1.5.0
Rigid transforms + Lie groups for JAX
29 versions - Latest release: 9 months ago - 300 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/explosion/thinc v9.0.0+incompatible
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
86 versions - Latest release: over 1 year ago - 2,878 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/tencent-quantum-lab/tensorcircuit v0.12.0
Tensor network based quantum software framework for the NISQ era
20 versions - Latest release: almost 2 years ago - 331 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/carnotresearch/cr-sparse v0.4.0 💰
Functional models and algorithms for sparse signal processing
10 versions - Latest release: about 2 years ago - 95 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/vicariousinc/pgmax v0.4.1
Loopy belief propagation for factor graphs on discrete variables, in JAX!
7 versions - Latest release: over 3 years ago - 63 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/mathpluscode/imgx-diffseg v0.3.2
A JAX-based deep learning framework for image segmentation using diffusion models.
4 versions - Latest release: about 2 years ago - 77 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/wandb/wandb v0.23.1
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models ...
168 versions - Latest release: about 1 month ago - 10,440 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/symjax/symjax v0.5.0
Documentation:
6 versions - Latest release: about 5 years ago - 121 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/google-deepmind/penzai v0.2.5
A JAX research toolkit for building, editing, and visualizing neural networks.
12 versions - Latest release: 9 months ago - 1,818 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/patrick-kidger/diffrax v0.7.0 💰
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs....
22 versions - Latest release: 10 months ago - 1,741 stars on GitHub
Top 8.2% on proxy.golang.org
github.com/tensorflow/datasets v4.9.9+incompatible
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
38 versions - Latest release: 7 months ago - 4,486 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/google/fedjax v0.0.17
FedJAX is a JAX-based open source library for Federated Learning simulations that emphasizes ease...
16 versions - Latest release: over 2 years ago - 266 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/patrick-kidger/lineax v0.0.8 💰
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
8 versions - Latest release: 10 months ago - 471 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/patrick-kidger/sympy2jax v0.0.7 💰
Turn SymPy expressions into trainable JAX expressions.
6 versions - Latest release: 9 months ago - 347 stars on GitHub
Top 5.9% on proxy.golang.org
github.com/patrick-kidger/optimistix v0.0.11 💰
Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.sit...
9 versions - Latest release: 3 months ago - 472 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/deepmind/pgmax v0.6.1
Loopy belief propagation for factor graphs on discrete variables in JAX
4 versions - Latest release: over 2 years ago - 150 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/google/neural-tangents v0.6.5
Fast and Easy Infinite Neural Networks in Python
17 versions - Latest release: about 2 years ago - 2,358 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/google/jaxtyping v0.3.4
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https...
46 versions - Latest release: 23 days ago - 1,010 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/sotetsuk/pgx v2.6.0+incompatible
♟️ Vectorized RL game environments in JAX
73 versions - Latest release: 10 months ago - 521 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/kubeflow/training-operator/v2 v2.1.0
Distributed ML Training and Fine-Tuning on Kubernetes
7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/mathpluscode/ImgX-DiffSeg v0.3.2
A JAX-based deep learning framework for image segmentation using diffusion models.
4 versions - Latest release: about 2 years ago - 77 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/deepmind/mctx v0.0.6
Monte Carlo tree search in JAX
5 versions - Latest release: 4 months ago - 2,214 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/kubeflow/tf-operator/v2 v2.1.0
Distributed ML Training and Fine-Tuning on Kubernetes
7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
Top 5.5% on proxy.golang.org
github.com/facebookresearch/fmmax v0.11.0
Fourier modal method with Jax
14 versions - Latest release: about 1 year ago - 115 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/nvidia/transformerengine v2.2.1+incompatible
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
2 versions - Latest release: 8 months ago - 2,867 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/nvidia/TransformerEngine v2.2.1+incompatible
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
2 versions - Latest release: 8 months ago - 2,840 stars on GitHub
Top 5.8% on proxy.golang.org
github.com/NVIDIA/TransformerEngine v2.2.1+incompatible
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
2 versions - Latest release: 8 months ago - 2,840 stars on GitHub
Top 4.2% on proxy.golang.org
github.com/arogozhnikov/einops v0.8.1
Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and ...
10 versions - Latest release: 11 months ago - 1 dependent repositories - 9,157 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/deepmodeling/deepmd-kit v3.1.1+incompatible
A deep learning package for many-body potential energy representation and molecular dynamics
66 versions - Latest release: 3 months ago - 1,655 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/ASEM000/kernex v0.2.1
Stencil computations in JAX
5 versions - Latest release: about 2 months ago - 62 stars on GitHub
Top 3.6% on proxy.golang.org
github.com/keras-team/keras v3.12.0+incompatible
Deep Learning for humans
59 versions - Latest release: 2 months ago - 1 dependent repositories - 63,497 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/kmheckel/spyx v0.1.20
Spyx: Spiking Neural Networks in JAX
2 versions - Latest release: over 1 year ago - 123 stars on GitHub
github.com/wandb/wandb/core v0.0.0-20240124235635-494df002f94d
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models ...
30 versions - Latest release: almost 2 years ago - 10,440 stars on GitHub
Top 5.2% on proxy.golang.org
github.com/wandb/wandb/nexus v0.0.0-20231123013222-99c2a2b5d26f
The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models ...
83 versions - Latest release: about 2 years ago - 10,440 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/DBraun/DawDreamer v0.8.3
Digital Audio Workstation with Python; VST instruments/effects, parameter automation, FAUST, JAX,...
28 versions - Latest release: over 1 year ago - 1,070 stars on GitHub
Top 9.6% on proxy.golang.org
github.com/alpa-projects/alpa v0.2.3
Training and serving large-scale neural networks with auto parallelization.
13 versions - Latest release: almost 3 years ago - 3,160 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/asem000/kernex v0.2.1
Stencil computations in JAX
5 versions - Latest release: about 2 months ago - 62 stars on GitHub
Top 5.7% on proxy.golang.org
github.com/google-deepmind/jraph v0.0.2-dev
A Graph Neural Network Library in Jax
1 version - Latest release: about 4 years ago - 1,443 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/alonfnt/bayex v0.2.2
Minimal Implementation of Bayesian Optimization in JAX
2 versions - Latest release: 12 months ago - 97 stars on GitHub
Top 6.7% on proxy.golang.org
github.com/pasqal-io/qadence v1.11.5
Digital-analog quantum programming interface
51 versions - Latest release: 7 months ago - 89 stars on GitHub
Top 5.6% on proxy.golang.org
github.com/bethgelab/foolbox v3.3.4+incompatible 💰
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow,...
50 versions - Latest release: almost 2 years ago - 2,911 stars on GitHub
Related Keywords
machine-learning 39 deep-learning 34 python 31 pytorch 30 tensorflow 27 reinforcement-learning 17 ai 15 neural-networks 11 mlops 10 flax 9 gpu 9 keras 7 data-science 7 numpy 7 hyperparameter-tuning 6 bayesian-inference 6 llm 6 cuda 6 reproducibility 5 fine-tuning 5 model-versioning 5 huggingface 5 collaboration 5 kubeflow 5 data-versioning 5 kubernetes 5 experiment-track 5 hyperparameter-optimization 5 hyperparameter-search 5 research 5 deep-reinforcement-learning 5 ml-platform 5 xgboost 4 jit 4 deep-neural-networks 4 equinox 4 nlp 4 automatic-differentiation 4 distributed 4 natural-language-processing 4 probabilistic-programming 4 transformer 4 quantum 4 quantum-computing 4 mlir 4 pde 3 differential-equations 3 gaussian-processes 3 speech-recognition 3 quantum-machine-learning 3 kernel 3 fp8 3 compiler 3 bert 3 hacktoberfest 3 language-model 3 language-models 3 model-hub 3 dataset 3 nlp-library 3 pretrained-models 3 pytorch-transformers 3 seq2seq 3 distributed-reinforcement-learning 2 end-to-end 2 pure-jax 2 reinforcement-learning-algorithms 2 sarl 2 sebulba 2 single-agent 2 single-agent-reinforcement-learning 2 single-agent-systems 2 theano 2 mxnet 2 neural-network 2 computational-chemistry 2 molecular-dynamics 2 unitaryhack 2 convex-optimization 2 optimization 2 artificial-intelligence 2 functional-programming 2 abdominal-organ-segmentation 2 brain-tumor-segmentation 2 diffusion-models 2 muscle-ultrasound-segmentation 2 prostate-segmentation 2 segmentation 2 stencil 2 graph-neural-networks 2 ai-for-science 2 benchmark 2 dae 2 differentialequations 2 julia 2 matlab 2 nerual-differential-equations 2 neural-ode 2 ode 2 partial-differential-equations 2