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
28 versions - Latest release: over 1 year ago - 1,070 stars on GitHub
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
9 versions - Latest release: about 1 month ago - 2,924 stars on GitHub
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
1 version - Latest release: over 1 year ago - 4 stars on GitHub
github.com/i-m-iron-man/foragax v0.0.5
An Agent-Based Modelling framework in JAX1 version - Latest release: over 1 year ago - 4 stars on GitHub
Top 6.7% on proxy.golang.org
13 versions - Latest release: 11 months ago - 353 stars on GitHub
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
49 versions - Latest release: 5 months ago - 496 stars on GitHub
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
Latest release: 1 day ago - 10,440 stars on GitHub
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
23 versions - Latest release: 3 months ago - 188 stars on GitHub
github.com/PennyLaneAI/catalyst v0.13.0
A JIT compiler for hybrid quantum programs in PennyLane23 versions - Latest release: 3 months ago - 188 stars on GitHub
Top 5.7% on proxy.golang.org
47 versions - Latest release: 2 days ago - 1,408 stars on GitHub
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
9 versions - Latest release: about 1 month ago - 3,426 stars on GitHub
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
7 versions - Latest release: 22 days ago - 298 stars on GitHub
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
1 version - Latest release: over 4 years ago - 584 stars on GitHub
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
11 versions - Latest release: 7 months ago - 318 stars on GitHub
github.com/adaptive-intelligent-robotics/QDax v0.5.0
Accelerated Quality-Diversity11 versions - Latest release: 7 months ago - 318 stars on GitHub
Top 6.7% on proxy.golang.org
3 versions - Latest release: almost 2 years ago - 200 stars on GitHub
github.com/google/tree-math v0.2.1
Mathematical operations for JAX pytrees3 versions - Latest release: almost 2 years ago - 200 stars on GitHub
Top 6.7% on proxy.golang.org
18 versions - Latest release: 7 months ago - 769 stars on GitHub
github.com/instadeepai/jumanji v1.1.1
🕹️ A diverse suite of scalable reinforcement learning environments in JAX18 versions - Latest release: 7 months ago - 769 stars on GitHub
Top 6.7% on proxy.golang.org
50 versions - Latest release: about 1 month ago - 387 stars on GitHub
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
50 versions - Latest release: about 1 month ago - 387 stars on GitHub
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
6 versions - Latest release: almost 6 years ago - 2,533 stars on GitHub
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
7 versions - Latest release: over 3 years ago - 63 stars on GitHub
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
36 versions - Latest release: 22 days ago - 2,886 stars on GitHub
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
23 versions - Latest release: 3 months ago - 188 stars on GitHub
github.com/pennylaneai/catalyst v0.13.0
A JIT compiler for hybrid quantum programs in PennyLane23 versions - Latest release: 3 months ago - 188 stars on GitHub
Top 6.7% on proxy.golang.org
11 versions - Latest release: 7 months ago - 319 stars on GitHub
github.com/adaptive-intelligent-robotics/qdax v0.5.0
Accelerated Quality-Diversity11 versions - Latest release: 7 months ago - 319 stars on GitHub
Top 7.6% on proxy.golang.org
Latest release: 5 days ago - 10,440 stars on GitHub
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
20 versions - Latest release: 21 days ago - 74 stars on GitHub
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
1 version - Latest release: over 3 years ago - 1 dependent repositories - 6,897 stars on GitHub
github.com/probml/pyprobml v0.1.0
Python code for "Probabilistic Machine learning" book by Kevin Murphy1 version - Latest release: over 3 years ago - 1 dependent repositories - 6,897 stars on GitHub
Top 6.7% on proxy.golang.org
13 versions - Latest release: 7 months ago - 20 stars on GitHub
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
7 versions - Latest release: 10 months ago - 34 stars on GitHub
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
1 version - Latest release: over 1 year ago - 5 stars on GitHub
github.com/i-m-iron-man/Foragax v0.0.5
An Agent-Based Modelling framework in JAX1 version - Latest release: over 1 year ago - 5 stars on GitHub
Top 6.7% on proxy.golang.org
69 versions - Latest release: about 1 month ago - 439 stars on GitHub
github.com/google/orbax v0.11.30
Orbax provides common checkpointing and persistence utilities for JAX users69 versions - Latest release: about 1 month ago - 439 stars on GitHub
Top 5.7% on proxy.golang.org
3 versions - Latest release: 4 months ago - 364 stars on GitHub
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
60 versions - Latest release: 20 days ago - 60,200 stars on GitHub
github.com/fchollet/keras v3.13.0+incompatible
Deep Learning for humans60 versions - Latest release: 20 days ago - 60,200 stars on GitHub
Top 9.6% on proxy.golang.org
232 versions - Latest release: about 1 month ago - 119,150 stars on GitHub
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
232 versions - Latest release: about 1 month ago - 126,450 stars on GitHub
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
232 versions - Latest release: about 1 month ago - 126,450 stars on GitHub
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
45 versions - Latest release: 6 months ago - 1,951 stars on GitHub
github.com/kubeflow/trainer v1.9.3
Distributed AI Model Training and Fine-Tuning on Kubernetes45 versions - Latest release: 6 months ago - 1,951 stars on GitHub
Top 5.5% on proxy.golang.org
7 versions - Latest release: 2 months ago - 1,951 stars on GitHub
github.com/kubeflow/trainer/v2 v2.1.0
Distributed AI Model Training and Fine-Tuning on Kubernetes7 versions - Latest release: 2 months ago - 1,951 stars on GitHub
Top 5.7% on proxy.golang.org
1 version - Latest release: about 4 years ago - 1,424 stars on GitHub
github.com/deepmind/jraph v0.0.2-dev
A Graph Neural Network Library in Jax1 version - Latest release: about 4 years ago - 1,424 stars on GitHub
Top 5.7% on proxy.golang.org
2 versions - Latest release: almost 3 years ago - 102 stars on GitHub
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
4 versions - Latest release: over 2 years ago - 330 stars on GitHub
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
3 versions - Latest release: 4 months ago - 364 stars on GitHub
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
3 versions - Latest release: 3 months ago - 26 stars on GitHub
github.com/bahremsd/tmmax v1.1.4
A fast transfer matrix method written in jax for modelling optical multilayer thin films3 versions - Latest release: 3 months ago - 26 stars on GitHub
Top 5.7% on proxy.golang.org
18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
github.com/deepmind/haiku v0.0.16
JAX-based neural network library18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
Top 5.7% on proxy.golang.org
6 versions - Latest release: about 3 years ago - 130 stars on GitHub
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
53 versions - Latest release: 3 months ago - 2,522 stars on GitHub
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
18 versions - Latest release: about 4 years ago - 1 dependent repositories - 8,287 stars on GitHub
github.com/google/trax v1.4.1
Trax — Deep Learning with Clear Code and Speed18 versions - Latest release: about 4 years ago - 1 dependent repositories - 8,287 stars on GitHub
Top 5.7% on proxy.golang.org
1 version - Latest release: over 4 years ago - 597 stars on GitHub
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
4 versions - Latest release: over 2 years ago - 150 stars on GitHub
github.com/deepmind/PGMax v0.6.1
Loopy belief propagation for factor graphs on discrete variables in JAX4 versions - Latest release: over 2 years ago - 150 stars on GitHub
Top 5.7% on proxy.golang.org
4 versions - Latest release: over 2 years ago - 332 stars on GitHub
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
6 versions - Latest release: about 5 years ago - 120 stars on GitHub
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
17 versions - Latest release: over 2 years ago - 11,842 stars on GitHub
github.com/unifyai/ivy v1.1.17
Unified AI17 versions - Latest release: over 2 years ago - 11,842 stars on GitHub
Top 5.6% on proxy.golang.org
2 versions - Latest release: about 2 months ago - 132 stars on GitHub
github.com/general-molecular-simulations/so3lr v0.1.0
SO3krates and Universal Pairwise Force Field for Molecular Simulation2 versions - Latest release: about 2 months ago - 132 stars on GitHub
Top 3.5% on proxy.golang.org
29 versions - Latest release: 2 months ago - 2 dependent repositories - 1,626 stars on GitHub
github.com/kubeflow/katib v0.19.0
Automated Machine Learning on Kubernetes29 versions - Latest release: 2 months ago - 2 dependent repositories - 1,626 stars on GitHub
Top 6.7% on proxy.golang.org
55 versions - Latest release: 24 days ago - 625 stars on GitHub
github.com/netket/netket v3.21.0+incompatible
Machine learning algorithms for many-body quantum systems55 versions - Latest release: 24 days ago - 625 stars on GitHub
Top 5.7% on proxy.golang.org
18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
github.com/deepmind/dm-haiku v0.0.16
JAX-based neural network library18 versions - Latest release: 21 days ago - 3,020 stars on GitHub
Top 6.7% on proxy.golang.org
7 versions - Latest release: 29 days ago - 141 stars on GitHub
github.com/lanl/scico v0.0.7
Scientific Computational Imaging COde7 versions - Latest release: 29 days ago - 141 stars on GitHub
Top 8.2% on proxy.golang.org
29 versions - Latest release: 9 months ago - 300 stars on GitHub
github.com/brentyi/jaxlie v1.5.0
Rigid transforms + Lie groups for JAX29 versions - Latest release: 9 months ago - 300 stars on GitHub
Top 5.6% on proxy.golang.org
86 versions - Latest release: over 1 year ago - 2,878 stars on GitHub
github.com/explosion/thinc v9.0.0+incompatible
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries86 versions - Latest release: over 1 year ago - 2,878 stars on GitHub
Top 6.7% on proxy.golang.org
20 versions - Latest release: almost 2 years ago - 331 stars on GitHub
github.com/tencent-quantum-lab/tensorcircuit v0.12.0
Tensor network based quantum software framework for the NISQ era20 versions - Latest release: almost 2 years ago - 331 stars on GitHub
Top 8.2% on proxy.golang.org
10 versions - Latest release: about 2 years ago - 95 stars on GitHub
github.com/carnotresearch/cr-sparse v0.4.0 💰
Functional models and algorithms for sparse signal processing10 versions - Latest release: about 2 years ago - 95 stars on GitHub
Top 6.7% on proxy.golang.org
7 versions - Latest release: over 3 years ago - 63 stars on GitHub
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
4 versions - Latest release: about 2 years ago - 77 stars on GitHub
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
168 versions - Latest release: about 1 month ago - 10,440 stars on GitHub
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
6 versions - Latest release: about 5 years ago - 121 stars on GitHub
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
12 versions - Latest release: 9 months ago - 1,818 stars on GitHub
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
22 versions - Latest release: 10 months ago - 1,741 stars on GitHub
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
38 versions - Latest release: 7 months ago - 4,486 stars on GitHub
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
16 versions - Latest release: over 2 years ago - 266 stars on GitHub
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
8 versions - Latest release: 10 months ago - 471 stars on GitHub
github.com/patrick-kidger/lineax v0.0.8 💰
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax8 versions - Latest release: 10 months ago - 471 stars on GitHub
Top 5.7% on proxy.golang.org
6 versions - Latest release: 9 months ago - 347 stars on GitHub
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
9 versions - Latest release: 3 months ago - 472 stars on GitHub
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
4 versions - Latest release: over 2 years ago - 150 stars on GitHub
github.com/deepmind/pgmax v0.6.1
Loopy belief propagation for factor graphs on discrete variables in JAX4 versions - Latest release: over 2 years ago - 150 stars on GitHub
Top 5.7% on proxy.golang.org
17 versions - Latest release: about 2 years ago - 2,358 stars on GitHub
github.com/google/neural-tangents v0.6.5
Fast and Easy Infinite Neural Networks in Python17 versions - Latest release: about 2 years ago - 2,358 stars on GitHub
Top 6.7% on proxy.golang.org
46 versions - Latest release: 23 days ago - 1,010 stars on GitHub
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
73 versions - Latest release: 10 months ago - 521 stars on GitHub
github.com/sotetsuk/pgx v2.6.0+incompatible
♟️ Vectorized RL game environments in JAX73 versions - Latest release: 10 months ago - 521 stars on GitHub
Top 5.5% on proxy.golang.org
7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
github.com/kubeflow/training-operator/v2 v2.1.0
Distributed ML Training and Fine-Tuning on Kubernetes7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
Top 5.5% on proxy.golang.org
4 versions - Latest release: about 2 years ago - 77 stars on GitHub
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
5 versions - Latest release: 4 months ago - 2,214 stars on GitHub
github.com/deepmind/mctx v0.0.6
Monte Carlo tree search in JAX5 versions - Latest release: 4 months ago - 2,214 stars on GitHub
Top 5.5% on proxy.golang.org
7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
github.com/kubeflow/tf-operator/v2 v2.1.0
Distributed ML Training and Fine-Tuning on Kubernetes7 versions - Latest release: 2 months ago - 1,821 stars on GitHub
Top 5.5% on proxy.golang.org
14 versions - Latest release: about 1 year ago - 115 stars on GitHub
github.com/facebookresearch/fmmax v0.11.0
Fourier modal method with Jax14 versions - Latest release: about 1 year ago - 115 stars on GitHub
Top 6.7% on proxy.golang.org
2 versions - Latest release: 8 months ago - 2,867 stars on GitHub
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
2 versions - Latest release: 8 months ago - 2,840 stars on GitHub
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
2 versions - Latest release: 8 months ago - 2,840 stars on GitHub
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
10 versions - Latest release: 11 months ago - 1 dependent repositories - 9,157 stars on GitHub
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
66 versions - Latest release: 3 months ago - 1,655 stars on GitHub
github.com/deepmodeling/deepmd-kit v3.1.1+incompatible
A deep learning package for many-body potential energy representation and molecular dynamics66 versions - Latest release: 3 months ago - 1,655 stars on GitHub
Top 6.7% on proxy.golang.org
5 versions - Latest release: about 2 months ago - 62 stars on GitHub
github.com/ASEM000/kernex v0.2.1
Stencil computations in JAX5 versions - Latest release: about 2 months ago - 62 stars on GitHub
Top 3.6% on proxy.golang.org
59 versions - Latest release: 2 months ago - 1 dependent repositories - 63,497 stars on GitHub
github.com/keras-team/keras v3.12.0+incompatible
Deep Learning for humans59 versions - Latest release: 2 months ago - 1 dependent repositories - 63,497 stars on GitHub
Top 6.7% on proxy.golang.org
2 versions - Latest release: over 1 year ago - 123 stars on GitHub
github.com/kmheckel/spyx v0.1.20
Spyx: Spiking Neural Networks in JAX2 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
83 versions - Latest release: about 2 years ago - 10,440 stars on GitHub
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
28 versions - Latest release: over 1 year ago - 1,070 stars on GitHub
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
13 versions - Latest release: almost 3 years ago - 3,160 stars on GitHub
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
5 versions - Latest release: about 2 months ago - 62 stars on GitHub
github.com/asem000/kernex v0.2.1
Stencil computations in JAX5 versions - Latest release: about 2 months ago - 62 stars on GitHub
Top 5.7% on proxy.golang.org
1 version - Latest release: about 4 years ago - 1,443 stars on GitHub
github.com/google-deepmind/jraph v0.0.2-dev
A Graph Neural Network Library in Jax1 version - Latest release: about 4 years ago - 1,443 stars on GitHub
Top 6.7% on proxy.golang.org
2 versions - Latest release: 12 months ago - 97 stars on GitHub
github.com/alonfnt/bayex v0.2.2
Minimal Implementation of Bayesian Optimization in JAX2 versions - Latest release: 12 months ago - 97 stars on GitHub
Top 6.7% on proxy.golang.org
51 versions - Latest release: 7 months ago - 89 stars on GitHub
github.com/pasqal-io/qadence v1.11.5
Digital-analog quantum programming interface51 versions - Latest release: 7 months ago - 89 stars on GitHub
Top 5.6% on proxy.golang.org
50 versions - Latest release: almost 2 years ago - 2,911 stars on GitHub
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